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

[cuda]
flash-attn>=2.8.2

[dev]
ruff
ipython
ipykernel
ipdb

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