torch
transformers
numpy
pytest
pyyaml
nltk
