torch
ninja
einops
transformers
packaging
setuptools>=61.0.0

[causal-conv1d]
causal-conv1d>=1.2.0

[dev]
pytest
