nvidia-cutlass-dsl==4.6.0.dev0
torch
apache-tvm-ffi<0.2,>=0.1.6
torch-c-dlpack-ext
einops

[bench]
pandas

[cu13]
nvidia-cutlass-dsl[cu13]==4.6.0.dev0

[dev]
pre-commit
pytest
pytest-xdist
ruff

[heuristics]
nvidia-matmul-heuristics

[jax]
jax
jax-tvm-ffi
