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

[dev]
