torch>=1.4.0
transformers>=4.18.0
numpy>=1.18.2
accelerate
datasets

[dev]
parameterized
pytest
pytest-xdist
black
isort
flake8>=3.8.3

[test]
parameterized
pytest
pytest-xdist
accelerate
