torch>=2.1.0

[test]
pytest>=1.3.2
scikit-learn>=1.3.2
numpy>=1.3.2
