jax<0.10,>=0.4.30
jaxlib>=0.4.30
flax<0.13,>=0.10
optax>=0.2
numpy
pyyaml
tokenizers
msgpack
tqdm

[all]
dantinox[benchmark,data,dev,elf,hub]

[benchmark]
pandas
matplotlib
numpy
scipy

[data]
datasets

[dev]
pytest>=8
pytest-cov
ruff
mypy
types-PyYAML
types-tqdm
interrogate>=1.7
pre-commit>=3
build
twine
mkdocs-material>=9
mkdocs-minify-plugin
mkdocs-glightbox
mkdocstrings[python]
mike

[elf]
transformers

[hub]
huggingface-hub
