torch>=2.1.0
numpy>=1.20.0
tqdm>=4.60.0
pyyaml>=6.0
transformers>=4.57.1
datasets>=2.0.0

[dev]
setuptools>=77.0
pytest>=7.0.0
black>=22.0.0
flake8>=4.0.0
mypy>=0.950
sphinx>=7.0.0
sphinx-markdown-builder>=0.6.0
build
twine

[docs]
sphinx>=7.0.0
sphinx-markdown-builder>=0.6.0

[wandb]
wandb>=0.16.0
