torch
transformers
numpy

[dev]
pip-tools
pytest
