torch
packaging
ninja
einops
triton
transformers
causal_conv1d<1.2.0,>=1.1.0
