torch>=2.0
numpy

[dev]
pytest>=7.0
ruff
