transformers
numpy
setuptools
torch