torch
tilelang==0.1.8
apache-tvm-ffi<=0.1.9
quack-kernels>=0.3.4
triton>=3.5.0
ninja
einops
transformers
packaging
setuptools>=61.0.0

[causal-conv1d]
causal-conv1d>=1.2.0

[dev]
pytest
