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
- chex >=0.1.8
- flax >=0.6.11
- jax >=0.4.25
- jaxlib >=0.4.20
- numpy >=1.19.5
- tensorstore >=0.1.51
- typing-extensions >=4.6.0
25 optional dependencies
-
black>=23.7.0[dev]extra == "dev"
-
coverage>=6.5.0[dev]extra == "dev"
-
distrax[examples]extra == "examples"
-
dm-haiku[examples]extra == "examples"
-
flake8>=6.0.0[dev]extra == "dev"
-
gymnasium[examples]extra == "examples"
-
isort>=5.12.0[dev]extra == "dev"
-
jumanji[examples]extra == "examples"
-
jupyterlab[examples]extra == "examples"
-
matplotlib[examples]extra == "examples"
-
mkdocs==1.5.3[dev]extra == "dev"
-
mkdocs-git-revision-date-plugin==0.3.2[dev]extra == "dev"
-
mkdocs-include-markdown-plugin==6.0.1[dev]extra == "dev"
-
mkdocs-material==9.4.2[dev]extra == "dev"
-
mkdocs-mermaid2-plugin==1.1.1[dev]extra == "dev"
-
mkdocstrings==0.23.0[dev]extra == "dev"
-
mknotebooks==0.8.0[dev]extra == "dev"
-
mypy>=1.8.0[dev]extra == "dev"
-
optax[examples]extra == "examples"
-
pre-commit>=2.20.0[dev]extra == "dev"
-
pytest>=7.4.2[dev]extra == "dev"
-
pytest-cov>=4.00[dev]extra == "dev"
-
pytest-xdist>=3.0.2[dev]extra == "dev"
-
rlax[examples]extra == "examples"
-
seaborn[examples]extra == "examples"