torch>=2.5
transformers>=4.45.0
triton>=3.0
datasets>=3.3.0
einops
ninja

[conv1d]
causal-conv1d>=1.4.0
