torch>=2.1
numpy

[test]
pytest
