torch>=2.1
transformers>=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

[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
