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

[conv1d]
causal-conv1d>=1.4.0

[dev]
pytest
