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.89
- diffrax >=0.6.2
- einops >=0.8.0
- flax >=0.12.1
- jax >=0.5
- jax-autovmap >=0.3.0
- jaxlib >=0.5
- numpy >=2.0.2
- optax >=0.2.4
24 optional dependencies
-
black>=23.0.0[dev]extra == "dev"
-
flowjax>=0.3.0[dev]extra == "dev"
-
furo[docs]extra == "docs"
-
hypothesis>=6.90[dev]extra == "dev"
-
ipykernel[docs]extra == "docs"
-
ipywidgets[docs]extra == "docs"
-
isort>=5.12.0[dev]extra == "dev"
-
matplotlib>=3.9.0[dev]extra == "dev"
-
matplotlib[docs]extra == "docs"
-
mypy>=1.0.0[dev]extra == "dev"
-
myst-nb[docs]extra == "docs"
-
nb-clean>=4.0.1[dev]extra == "dev"
-
numpy[docs]extra == "docs"
-
numpydoc[docs]extra == "docs"
-
pre-commit>=3.0.0[dev]extra == "dev"
-
pytest>=7.0[dev]extra == "dev"
-
pytest-cov>=4.0.0[dev]extra == "dev"
-
pytest-xdist>=3.0.0[dev]extra == "dev"
-
ruff>=0.0.241[dev]extra == "dev"
-
sphinx[docs]extra == "docs"
-
sphinx-autobuild[docs]extra == "docs"
-
sphinx-autodoc-typehints[docs]extra == "docs"
-
sphinx-copybutton[docs]extra == "docs"
-
sphinx-math-dollar[docs]extra == "docs"