torch>=2.1

[test]
pytest>=7
numpy>=1.21
