numpy>=1.20
scikit-learn>=1.0
scipy>=1.7

[dev]
pytest>=7.0
pytest-cov

[gpu]
torch>=2.0
