torch
packaging
ninja
einops
triton
transformers
causal_conv1d>=1.1.0
