torch>=2.6
typer
aiofiles
deprecated
numpy
datasets
transformers>=4.57.1
ninja
numba>=0.61.2
rich

[cuda]
flash-attn>=2.8.2
liger-kernel>=0.5.10
kernels
mamba-ssm[causal-conv1d]>=2.2.5

[dev]
ruff
ipython
ipykernel
ipdb
pytest
tox
tox-uv

[mlflow]
mlflow>=3.0

[test]
pytest>=7.0
pytest-cov
pytest-mock

[wandb]
wandb
