jax
jaxlib
flax
optax
pyyaml
tokenizers
msgpack
tqdm

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

[benchmark]
pandas
matplotlib
numpy
scipy

[data]
datasets

[dev]
pytest>=8
pytest-cov
ruff
mypy
types-PyYAML
build
twine
mkdocs-material
mkdocs-minify-plugin
mkdocs-glightbox
mkdocstrings[python]

[hub]
huggingface-hub
