nvidia-cutlass-dsl>=4.4.1
torch
apache-tvm-ffi<0.2,>=0.1.6
torch-c-dlpack-ext

[cu13]
nvidia-cutlass-dsl[cu13]>=4.4.1

[dev]
pre-commit
pytest
pytest-xdist
ruff
