click<9.0,>=8.0.4
datasets<4.1,>=3.0
filelock
jsonlines
lightning<2.6,>=2.5.3
loguru
pydantic
pytorch-lightning<2.6,>=2.5.3
rich<15,>=14
sacrebleu
sacremoses
system_info
tensorboardx
torchmetrics<2.0,>=1.0
tokenizers<0.22,>=0.10
tomli
torch<2.9,>=2.0
transformers<5.0,>=4.55

[lint]
mypy
ruff

[tests]
pytest
pytest-console-scripts
pytest-github-actions-annotate-failures
