torch>=2.8.0
numpy
packaging

[dev]
pytest
pytest-benchmark
