torch>=2.0
cuda-python>=12.0

[dev]
triton>=3.0
pytest
mpmath

[triton]
triton>=3.0
