torch>=2.1
transformers>=4.36
safetensors>=0.4
rich>=13.0
typer>=0.9
Pillow>=10.0
matplotlib>=3.8
huggingface-hub>=0.20

[dev]
pytest>=7.0
pytest-timeout>=2.2
scikit-learn>=1.3
torchvision>=0.16

[probe]
scikit-learn>=1.3

[vision]
torchvision>=0.16
