torch>=2.0
numpy>=1.21

[dev]
pytest>=7
pytest-cov

[sklearn]
scikit-learn>=1.0
