einops>=0.7.0
huggingface_hub>=0.24.7
jaxtyping>=0.2.28
numpy
scipy
tokenizers>=0.15.2

[dev]
pre-commit
pytest
ruff

[docs]
furo
myst-parser>=1.0
sphinx-copybutton
sphinx-design
sphinxcontrib-mermaid
sphinx>=7.2

[torch]
torch>=2.4.0
huggingface_hub[torch]>=0.24.7
