transformers
numpy
setuptools
torch
bfloat16
sentencepiece