torch>=2.6
typer
aiofiles
deprecated
numpy<2.3
datasets
transformers>=4.55.0
ninja
numba
rich
pytest
tox
tox-uv

[cuda]
flash-attn>=2.8.2
liger-kernel>=0.5.10
kernels

[dev]
ruff
ipython
ipykernel
ipdb

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

[wandb]
wandb
