torch>=2.2
numpy

[dev]
pytest
