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
- jax >=0.11.1
- msgpack
- numpy >=1.23.2
- optax
- orbax-checkpoint
- pyyaml >=5.4.1
- rich >=11.1
- tensorstore
- treescope >=0.1.7
- typing-extensions >=4.2
51 optional dependencies
-
ale-py>=0.10.2[testing]python_version < "3.14" and extra == "testing"
-
cloudpickle>=3.0.0[testing]extra == "testing"
-
clu[testing]extra == "testing"
-
dm-haiku>=0.0.14[docs]extra == "docs"
-
docutils[docs]extra == "docs"
-
einops[docs]extra == "docs"
-
einops[testing]extra == "testing"
-
grain[docs]extra == "docs"
-
grain==0.2.16[testing]extra == "testing"
-
gymnasium[testing]python_version < "3.14" and extra == "testing"
-
ipykernel[docs]extra == "docs"
-
ipython-genutils[docs]extra == "docs"
-
ipywidgets>=8.1.5[docs]extra == "docs"
-
jaxlib[testing]extra == "testing"
-
jaxtyping[testing]extra == "testing"
-
jaxtyping[docs]extra == "docs"
-
jraph>=0.0.6dev0[testing]extra == "testing"
-
jupytext==1.13.8[docs]extra == "docs"
-
kagglehub>=0.3.3[docs]extra == "docs"
-
keras<3.13[testing]extra == "testing"
-
matplotlib[docs]extra == "docs"
-
ml-collections[testing]extra == "testing"
-
ml-collections[docs]extra == "docs"
-
mypy[testing]extra == "testing"
-
myst-nb[docs]extra == "docs"
-
nanobind>=2.5.0[dev]extra == "dev"
-
nbstripout[docs]extra == "docs"
-
opencv-python[testing]extra == "testing"
-
orbax-export>=0.0.8[docs]extra == "docs"
-
pre-commit>=3.8.0[dev]extra == "dev"
-
protobuf[testing]extra == "testing"
-
pygments>=2.6.1[docs]extra == "docs"
-
pytest[testing]extra == "testing"
-
pytest-cov[testing]extra == "testing"
-
pytest-custom-exit-code[testing]extra == "testing"
-
pytest-xdist[testing]extra == "testing"
-
pytype[testing]extra == "testing"
-
recommonmark[docs]extra == "docs"
-
scikit-build-core>=0.11.0[dev]extra == "dev"
-
scikit-learn[docs]extra == "docs"
-
sentencepiece==0.2.0[testing]extra == "testing"
-
sphinx==6.2.1[docs]extra == "docs"
-
sphinx-book-theme[docs]extra == "docs"
-
sphinx-design[docs]extra == "docs"
-
tensorboard[testing]extra == "testing"
-
tensorflow>=2.12.0[testing]python_version < "3.13" and extra == "testing"
-
tensorflow-datasets[testing]extra == "testing"
-
tensorflow-text>=2.11.0[testing](platform_system != "Darwin" and python_version < "3.13") and extra == "testing"
-
torch[testing]extra == "testing"
-
tqdm==4.67.1[docs]extra == "docs"
-
treescope>=0.1.1[testing]python_version >= "3.10" and extra == "testing"
Used By
- qwix 609.8K/mo
- aqtp 492.0K/mo
- clu 436.4K/mo
- google-tunix 263.0K/mo
- tensorflowjs 173.8K/mo
- gymnax 76.4K/mo
- google-metrax 53.4K/mo
- aisimulate 43.6K/mo
- dopamine-rl 43.2K/mo
- brax 41.1K/mo