supervision~=0.24.0rc1
requests<=2.32.3,>=2.31.0
transformers~=4.44.2
torch~=2.4.0
accelerate<0.35,>=0.33
sentencepiece~=0.2.0
peft~=0.12.0
einops~=0.8.0
timm~=1.0.9
typer~=0.12.5
jiwer~=3.0.4

[:sys_platform != "darwin"]
flash-attn~=2.6.3

[dev]
pytest~=8.3.2
pre-commit~=3.8.0
mypy~=1.11.2
ruff~=0.6.5
tox<4.21.0,>=4.18.1

[docs]
mkdocs-material~=9.5.35
mkdocstrings[python]<0.26.2,>=0.20.0
mike~=2.1.3
