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
Weekly Downloads
GitHub Stars
Dependencies
- beartype >=0.21.0
- black >=25.1.0
- chex >=0.1.85
- cmocean >=4.0.3
- ipykernel >=6.29.5
-
jax
>=0.7.0
sys_platform == "win64"
- jaxtyping >0.3.2
- jupyter >=1.1.1
- jupyterlab >=4.4.5
- matplotlib >=2.2.0
- matplotlib-scalebar >=0.5.0
- numpy >=1.13.0
- optax >=0.1.8
- psutil >=5.9.0
- pytest >=8.0.0
- pytest-cov >=6.2.1
- scipy >=1.0.0
- tqdm >=4.66.0
- xprof >=2.21.6
35 optional dependencies
-
black>=25.1.0[dev]extra == "dev"
-
chex>=0.1.89[test]extra == "test"
-
furo>=2025.7.19[docs]extra == "docs"
-
interrogate>=1.7.0[docs]extra == "docs"
-
ipykernel>=6.29.5[docs]extra == "docs"
-
ipykernel>=6.29.5[notebooks]extra == "notebooks"
-
ipykernel>=6.29.5[dev]extra == "dev"
-
ipywidgets>=8.1.0[notebooks]extra == "notebooks"
-
isort>=6.0.1[dev]extra == "dev"
-
janssen[dev-cuda]extra == "dev-cuda"
-
janssen[dev]extra == "dev"
-
jax>=0.7.0[cuda]platform_machine == "x86_64" and sys_platform == "linux" and extra == "cuda"
-
myst-parser>=4.0.1[docs]extra == "docs"
-
myst-parser>=4.0.1[dev]extra == "dev"
-
nbconvert>=7.16.6[notebooks]extra == "notebooks"
-
nbsphinx>=0.9.6[docs]extra == "docs"
-
nbsphinx>=0.9.6[dev]extra == "dev"
-
nvidia-cudnn-cu12>=9.5.0[cuda]platform_machine == "x86_64" and sys_platform == "linux" and extra == "cuda"
-
pandoc>=2.4[docs]extra == "docs"
-
pydata-sphinx-theme>=0.16.1[dev]extra == "dev"
-
pydata-sphinx-theme>=0.16.1[docs]extra == "docs"
-
pydoclint>=0.7.3[docs]extra == "docs"
-
pygount>=3.1.0[dev]extra == "dev"
-
pytest>=8.3.5[test]extra == "test"
-
pytest-cov>=4.0.0[test]extra == "test"
-
pytest-xdist>=3.0.0[test]extra == "test"
-
ruff>=0.12.9[dev]extra == "dev"
-
sphinx>=8.1.3[docs]extra == "docs"
-
sphinx>=8.1.3[dev]extra == "dev"
-
sphinx-autodoc-typehints>=3.0.1[docs]extra == "docs"
-
sphinx-rtd-theme>=3.0.2[docs]extra == "docs"
-
sphinx-rtd-theme>=3.0.2[dev]extra == "dev"
-
sphinxcontrib-napoleon>=0.7[dev]extra == "dev"
-
sphinxcontrib-napoleon>=0.7[docs]extra == "docs"
-
twine>=6.1.0[dev]extra == "dev"