numpy==2.2.6
rich==15.0.0
safetensors==0.8.0
scikit-learn==1.7.2
scipy==1.15.3
torch==2.10.0
torchmetrics==1.9.0
