torch>=2.2.0
ninja

[dev]
pytest
ruff
black
