torch>=2.0
numpy>=1.22
scipy>=1.9
scikit-learn>=1.1
joblib>=1.2
h5py>=3.7

[dev]
pytest>=7
