torch
ninja
einops
triton
transformers
packaging
setuptools>=61.0.0

[causal-conv1d]
causal-conv1d>=1.2.0

[dev]
pytest
