torch>=2.0
numpy>=1.24
scipy>=1.10
einops>=0.7
safetensors>=0.4
scikit-learn>=1.3
matplotlib