Downloads in the 30 and 7 days to October 10, 2026. Refreshed weekly.
Package Information
No reported malware or high/critical advisories · Details
The current version has no OSV.dev malware reports and no critical or high-severity vulnerabilities.
Package Insights
((week_daily_avg - month_daily_avg) / month_daily_avg) * 100More insights and metrics
Downloads by OS
Python Versions
Top Countries
Weekly Downloads
GitHub Stars
Dependencies
- flax <0.13,>=0.10
- jax <0.10,>=0.4.30
- jaxlib >=0.4.30
- msgpack
- numpy
- optax >=0.2
- pyyaml
- tokenizers
- tqdm
23 optional dependencies
-
build[dev]extra == "dev"
-
dantinox[all]extra == "all"
-
datasets[data]extra == "data"
-
huggingface-hub[hub]extra == "hub"
-
interrogate>=1.7[dev]extra == "dev"
-
matplotlib[benchmark]extra == "benchmark"
-
mike[dev]extra == "dev"
-
mkdocs-glightbox[dev]extra == "dev"
-
mkdocs-material>=9[dev]extra == "dev"
-
mkdocs-minify-plugin[dev]extra == "dev"
-
mkdocstrings[dev]extra == "dev"
-
mypy[dev]extra == "dev"
-
numpy[benchmark]extra == "benchmark"
-
pandas[benchmark]extra == "benchmark"
-
pre-commit>=3[dev]extra == "dev"
-
pytest>=8[dev]extra == "dev"
-
pytest-cov[dev]extra == "dev"
-
ruff[dev]extra == "dev"
-
scipy[benchmark]extra == "benchmark"
-
transformers[elf]extra == "elf"
-
twine[dev]extra == "dev"
-
types-pyyaml[dev]extra == "dev"
-
types-tqdm[dev]extra == "dev"