torch>=2.0.0
numpy

[dev]
pytest
pytest-cov
