torch>=2.0

[dev]
pytest

[gpu]
triton
