nltk
numpy
pytest
pyyaml
torch
transformers
