torch>=2.0
numpy>=1.24

[dev]
pytest
