transformers
datasets
einops
ninja

[conv1d]
causal-conv1d>=1.4.0

[cuda]
triton

[rocm]
pytorch-triton-rocm

[xpu]
pytorch-triton-xpu
