torch>=2.1
numpy>=1.24
transformers<6,>=4.36
safetensors>=0.4
rich>=13.0
rich-gradient>=0.3
typer>=0.9
Pillow>=10.0
matplotlib>=3.8
huggingface-hub>=0.20
sentencepiece>=0.1.99

[data]
datasets>=2.14

[dev]
pytest>=7.0
pytest-timeout>=2.2
pytest-cov>=5.0
scikit-learn>=1.3
torchvision>=0.16
ruff>=0.4
mypy>=1.8

[docs]
mkdocs>=1.5
mkdocs-material>=9.5
mkdocstrings[python]>=0.24

[probe]
scikit-learn>=1.3

[vision]
torchvision>=0.16
