torch>=2.6
typer
aiofiles
deprecated
numpy<2
datasets
transformers>=4.55.0
ninja
numba
rich
pytest
tox
tox-uv
liger-kernel
kernels

[cuda]
flash-attn>=2.8.2

[dev]
ruff
ipython
ipykernel
ipdb

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

[wandb]
wandb
