torch<2.8,>=2.0
numpy>=1.20

[bench]
matplotlib
scikit-learn

[test]
pytest
scikit-learn
