torch>=2.1.0
torchvision>=0.16
transformers[torch]>=4.46.1
accelerate>=0.26.0
scipy>=1.3.1
scikit-learn>=1.7
scikit-image>0.23.1
numpy<3.0.0,>=2.0.0
umap-learn>=0.5.5
timm
pandas
matplotlib
cmasher
tqdm
requests
pillow
captum>=0.7.0
ipywidgets
httpx
pyarrow

[dev]
pytest>=8.3.5
pytest-cov==6.3.0
