python>=3.9
numpy
scipy
torch>=1.7
tqdm
transformers
typing_extensions
