triton>=2.0.0

[cute]
nvidia-cutlass-dsl>=4.4.2
einops
typing_extensions
apache-tvm-ffi<0.2,>=0.1.5
torch-c-dlpack-ext
quack-kernels>=0.3.3

[dev]
transformers>=4.38.0
pytest>=6.0
pytest-benchmark
numpy
tabulate>=0.9.0

[docs]
mkdocs>=1.6
mkdocs-material>=9.5
mkdocstrings[python]>=0.28
pymdown-extensions>=10.0
