numpy
transformers
torch