torch>=2.1
torchvision>=0.16
transformers>=5.0
accelerate>=0.30
pandas>=2.0
numpy>=1.24
matplotlib>=3.8
pyyaml>=5.1

[dev]
pytest>=7.0
pytest-cov>=4.0
mypy>=1.0
ruff>=0.1
jsonschema>=4.18

[docs]
mkdocs>=1.6

[models]
tiktoken>=0.6
einops>=0.7
