torch>=2.0.0
numpy>=1.21

[test]
pytest>=7
