torch<1.12.0,>=1.10.0
pyyaml>=5.1
numpy<2.0.0,>1.16.0
sacrebleu==1.4.14

[dev]
pytest
pytest-cov<2.6.0
pillow
check-manifest
matplotlib
mypy==0.790
pylint==2.6.0
isort<5.0.0
setuptools>=38.6.0
twine>=1.11.0
wheel>=0.31.0

[optional]
tensorboard
matplotlib
