torch
packaging
ninja
einops
triton
transformers
causal_conv1d
