torch>=2.0

[cubin]
cuda-python>=12.0

[dev]
triton>=3.0
pytest
mpmath

[triton]
triton>=3.0
