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
- click >=8.0
- flax >=0.12.9
- jax >=0.11.1
- jaxlib >=0.11.1
- jaxtyping >=0.2.20
- lazy-loader >=0.5
- numpy >=2.1
- psutil >=5.9.5
- pydantic >=2.10
- scipy >=1.15
- substrax >=0.1.15
273 optional dependencies
-
bandit>=1.8.6[all-cpu]extra == "all-cpu"
-
bandit>=1.8.6[all]extra == "all"
-
bandit>=1.8.6[cuda-dev]extra == "cuda-dev"
-
bandit>=1.8.6[dev]extra == "dev"
-
bandit>=1.8.6[all-macos]extra == "all-macos"
-
beartype>=0.14.1[all]extra == "all"
-
beartype>=0.14.1[all-cpu]extra == "all-cpu"
-
beartype>=0.14.1[all-macos]extra == "all-macos"
-
beartype>=0.14.1[test]extra == "test"
-
build>=1.0.3[cuda-dev]extra == "cuda-dev"
-
build>=1.0.3[dev]extra == "dev"
-
build>=1.0.3[all]extra == "all"
-
build>=1.0.3[all-cpu]extra == "all-cpu"
-
build>=1.0.3[all-macos]extra == "all-macos"
-
codecarbon>=2.0[codecarbon]extra == "codecarbon"
-
codecarbon>=2.0[test]extra == "test"
-
codecarbon>=2.0[all]extra == "all"
-
codecarbon>=2.0[all-cpu]extra == "all-cpu"
-
codecarbon>=2.0[all-macos]extra == "all-macos"
-
coverage>=7[all]extra == "all"
-
coverage>=7[all-cpu]extra == "all-cpu"
-
coverage>=7[all-macos]extra == "all-macos"
-
coverage>=7[cuda-dev]extra == "cuda-dev"
-
coverage>=7[dev]extra == "dev"
-
coverage>=7[test]extra == "test"
-
griffe>=1.7.3[all]extra == "all"
-
griffe>=1.7.3[docs]extra == "docs"
-
griffe>=1.7.3[all-macos]extra == "all-macos"
-
griffe>=1.7.3[all-cpu]extra == "all-cpu"
-
hippogriffe>=0.2[docs]extra == "docs"
-
hippogriffe>=0.2[all-macos]extra == "all-macos"
-
hippogriffe>=0.2[all-cpu]extra == "all-cpu"
-
hippogriffe>=0.2[all]extra == "all"
-
hypothesis>=6.100[all-cpu]extra == "all-cpu"
-
hypothesis>=6.100[all]extra == "all"
-
hypothesis>=6.100[all-macos]extra == "all-macos"
-
hypothesis>=6.100[test]extra == "test"
-
import-linter>=2.0[dev]extra == "dev"
-
import-linter>=2.0[all]extra == "all"
-
import-linter>=2.0[all-cpu]extra == "all-cpu"
-
import-linter>=2.0[all-macos]extra == "all-macos"
-
import-linter>=2.0[cuda-dev]extra == "cuda-dev"
-
interrogate>=1.5[all-macos]extra == "all-macos"
-
interrogate>=1.5[cuda-dev]extra == "cuda-dev"
-
interrogate>=1.5[dev]extra == "dev"
-
interrogate>=1.5[all-cpu]extra == "all-cpu"
-
interrogate>=1.5[all]extra == "all"
-
ipykernel>=6.29.5[dev]extra == "dev"
-
ipykernel>=6.29.5[all]extra == "all"
-
ipykernel>=6.29.5[all-macos]extra == "all-macos"
-
ipykernel>=6.29.5[all-cpu]extra == "all-cpu"
-
ipykernel>=6.29.5[cuda-dev]extra == "cuda-dev"
-
jax>=0.11.1[cuda12]extra == "cuda12"
-
jax>=0.11.1[cuda-dev]extra == "cuda-dev"
-
jax>=0.11.1[all]extra == "all"
-
jax-metal>=0.1.0[all-macos]extra == "all-macos"
-
jax-metal>=0.1.0[metal]extra == "metal"
-
jupytext>=1.16.6[dev]extra == "dev"
-
jupytext>=1.16.6[cuda-dev]extra == "cuda-dev"
-
jupytext>=1.16.6[all-macos]extra == "all-macos"
-
jupytext>=1.16.6[all-cpu]extra == "all-cpu"
-
jupytext>=1.16.6[all]extra == "all"
-
matplotlib>=3.8[test]extra == "test"
-
matplotlib>=3.8[all]extra == "all"
-
matplotlib>=3.8[all-cpu]extra == "all-cpu"
-
matplotlib>=3.8[all-macos]extra == "all-macos"
-
matplotlib>=3.8[publication]extra == "publication"
-
mkdocs<2.0,>=1.6.1[all-cpu]extra == "all-cpu"
-
mkdocs<2.0,>=1.6.1[all-macos]extra == "all-macos"
-
mkdocs<2.0,>=1.6.1[docs]extra == "docs"
-
mkdocs<2.0,>=1.6.1[all]extra == "all"
-
mkdocs-include-exclude-files>=0.1[all-cpu]extra == "all-cpu"
-
mkdocs-include-exclude-files>=0.1[all-macos]extra == "all-macos"
-
mkdocs-include-exclude-files>=0.1[all]extra == "all"
-
mkdocs-include-exclude-files>=0.1[docs]extra == "docs"
-
mkdocs-ipynb>=0.1[all-cpu]extra == "all-cpu"
-
mkdocs-ipynb>=0.1[all-macos]extra == "all-macos"
-
mkdocs-ipynb>=0.1[docs]extra == "docs"
-
mkdocs-ipynb>=0.1[all]extra == "all"
-
mkdocs-jupyter>=0.24[all-cpu]extra == "all-cpu"
-
mkdocs-jupyter>=0.24[all-macos]extra == "all-macos"
-
mkdocs-jupyter>=0.24[docs]extra == "docs"
-
mkdocs-jupyter>=0.24[all]extra == "all"
-
mkdocs-material>=9.6.7[all-cpu]extra == "all-cpu"
-
mkdocs-material>=9.6.7[all-macos]extra == "all-macos"
-
mkdocs-material>=9.6.7[docs]extra == "docs"
-
mkdocs-material>=9.6.7[all]extra == "all"
-
mkdocstrings>=0.28.3[all-cpu]extra == "all-cpu"
-
mkdocstrings>=0.28.3[all-macos]extra == "all-macos"
-
mkdocstrings>=0.28.3[docs]extra == "docs"
-
mkdocstrings>=0.28.3[all]extra == "all"
-
mkdocstrings-python>=1.1.2[all]extra == "all"
-
mkdocstrings-python>=1.1.2[all-macos]extra == "all-macos"
-
mkdocstrings-python>=1.1.2[docs]extra == "docs"
-
mkdocstrings-python>=1.1.2[all-cpu]extra == "all-cpu"
-
mlflow>=2.0[test]extra == "test"
-
mlflow>=2.0[all-cpu]extra == "all-cpu"
-
mlflow>=2.0[all]extra == "all"
-
mlflow>=2.0[mlflow]extra == "mlflow"
-
mlflow>=2.0[all-macos]extra == "all-macos"
-
nvidia-cublas-cu12>=12.6[all]extra == "all"
-
nvidia-cublas-cu12>=12.6[cuda-dev]extra == "cuda-dev"
-
nvidia-cublas-cu12>=12.6[cuda12]extra == "cuda12"
-
nvidia-cuda-cupti-cu12>=12.6[cuda-dev]extra == "cuda-dev"
-
nvidia-cuda-cupti-cu12>=12.6[all]extra == "all"
-
nvidia-cuda-cupti-cu12>=12.6[cuda12]extra == "cuda12"
-
nvidia-cuda-nvcc-cu12>=12.6[cuda-dev]extra == "cuda-dev"
-
nvidia-cuda-nvcc-cu12>=12.6[cuda12]extra == "cuda12"
-
nvidia-cuda-nvcc-cu12>=12.6[all]extra == "all"
-
nvidia-cuda-nvrtc-cu12>=12.6[cuda-dev]extra == "cuda-dev"
-
nvidia-cuda-nvrtc-cu12>=12.6[cuda12]extra == "cuda12"
-
nvidia-cuda-nvrtc-cu12>=12.6[all]extra == "all"
-
nvidia-cuda-runtime-cu12>=12.6[cuda12]extra == "cuda12"
-
nvidia-cuda-runtime-cu12>=12.6[all]extra == "all"
-
nvidia-cuda-runtime-cu12>=12.6[cuda-dev]extra == "cuda-dev"
-
nvidia-cudnn-cu12>=9.1[cuda-dev]extra == "cuda-dev"
-
nvidia-cudnn-cu12>=9.1[cuda12]extra == "cuda12"
-
nvidia-cudnn-cu12>=9.1[all]extra == "all"
-
nvidia-cufft-cu12>=11.4[cuda-dev]extra == "cuda-dev"
-
nvidia-cufft-cu12>=11.4[cuda12]extra == "cuda12"
-
nvidia-cufft-cu12>=11.4[all]extra == "all"
-
nvidia-curand-cu12>=10.3[cuda-dev]extra == "cuda-dev"
-
nvidia-curand-cu12>=10.3[cuda12]extra == "cuda12"
-
nvidia-curand-cu12>=10.3[all]extra == "all"
-
nvidia-cusolver-cu12>=11.7[cuda-dev]extra == "cuda-dev"
-
nvidia-cusolver-cu12>=11.7[cuda12]extra == "cuda12"
-
nvidia-cusolver-cu12>=11.7[all]extra == "all"
-
nvidia-cusparse-cu12>=12.3[cuda-dev]extra == "cuda-dev"
-
nvidia-cusparse-cu12>=12.3[cuda12]extra == "cuda12"
-
nvidia-cusparse-cu12>=12.3[all]extra == "all"
-
nvidia-ml-py>=12.0.0[cuda-dev]extra == "cuda-dev"
-
nvidia-ml-py>=12.0.0[all]extra == "all"
-
nvidia-ml-py>=12.0.0[cuda12]extra == "cuda12"
-
nvidia-nccl-cu12>=2.20[cuda12]extra == "cuda12"
-
nvidia-nccl-cu12>=2.20[all]extra == "all"
-
nvidia-nccl-cu12>=2.20[cuda-dev]extra == "cuda-dev"
-
nvidia-nvjitlink-cu12>=12.6[all]extra == "all"
-
nvidia-nvjitlink-cu12>=12.6[cuda12]extra == "cuda12"
-
nvidia-nvjitlink-cu12>=12.6[cuda-dev]extra == "cuda-dev"
-
nvidia-nvtx-cu12>=12.6[all]extra == "all"
-
nvidia-nvtx-cu12>=12.6[cuda-dev]extra == "cuda-dev"
-
nvidia-nvtx-cu12>=12.6[cuda12]extra == "cuda12"
-
pip-audit>=2.7[all-macos]extra == "all-macos"
-
pip-audit>=2.7[cuda-dev]extra == "cuda-dev"
-
pip-audit>=2.7[dev]extra == "dev"
-
pip-audit>=2.7[all]extra == "all"
-
pip-audit>=2.7[all-cpu]extra == "all-cpu"
-
pre-commit>=4.2[dev]extra == "dev"
-
pre-commit>=4.2[all-cpu]extra == "all-cpu"
-
pre-commit>=4.2[all]extra == "all"
-
pre-commit>=4.2[all-macos]extra == "all-macos"
-
pre-commit>=4.2[cuda-dev]extra == "cuda-dev"
-
pydeps>=1.12.17[all-macos]extra == "all-macos"
-
pydeps>=1.12.17[cuda-dev]extra == "cuda-dev"
-
pydeps>=1.12.17[dev]extra == "dev"
-
pydeps>=1.12.17[all]extra == "all"
-
pydeps>=1.12.17[all-cpu]extra == "all-cpu"
-
pydoclint>=0.5[all-macos]extra == "all-macos"
-
pydoclint>=0.5[cuda-dev]extra == "cuda-dev"
-
pydoclint>=0.5[dev]extra == "dev"
-
pydoclint>=0.5[all]extra == "all"
-
pydoclint>=0.5[all-cpu]extra == "all-cpu"
-
pygments<3.0,>=2.20[all-macos]extra == "all-macos"
-
pygments<3.0,>=2.20[cuda-dev]extra == "cuda-dev"
-
pygments<3.0,>=2.20[dev]extra == "dev"
-
pygments<3.0,>=2.20[docs]extra == "docs"
-
pygments<3.0,>=2.20[all]extra == "all"
-
pygments<3.0,>=2.20[all-cpu]extra == "all-cpu"
-
pymdown-extensions>=10.16.1[all-macos]extra == "all-macos"
-
pymdown-extensions>=10.16.1[docs]extra == "docs"
-
pymdown-extensions>=10.16.1[all]extra == "all"
-
pymdown-extensions>=10.16.1[all-cpu]extra == "all-cpu"
-
pyright>=1.1.336[all]extra == "all"
-
pyright>=1.1.336[all-cpu]extra == "all-cpu"
-
pyright>=1.1.336[all-macos]extra == "all-macos"
-
pyright>=1.1.336[cuda-dev]extra == "cuda-dev"
-
pyright>=1.1.336[dev]extra == "dev"
-
pytest>=9.0.3[all-macos]extra == "all-macos"
-
pytest>=9.0.3[cuda-dev]extra == "cuda-dev"
-
pytest>=9.0.3[dev]extra == "dev"
-
pytest>=9.0.3[test]extra == "test"
-
pytest>=9.0.3[all-cpu]extra == "all-cpu"
-
pytest>=9.0.3[all]extra == "all"
-
pytest-asyncio>=0.23[all-macos]extra == "all-macos"
-
pytest-asyncio>=0.23[cuda-dev]extra == "cuda-dev"
-
pytest-asyncio>=0.23[dev]extra == "dev"
-
pytest-asyncio>=0.23[test]extra == "test"
-
pytest-asyncio>=0.23[all]extra == "all"
-
pytest-asyncio>=0.23[all-cpu]extra == "all-cpu"
-
pytest-benchmark>=4[all-macos]extra == "all-macos"
-
pytest-benchmark>=4[cuda-dev]extra == "cuda-dev"
-
pytest-benchmark>=4[dev]extra == "dev"
-
pytest-benchmark>=4[test]extra == "test"
-
pytest-benchmark>=4[all]extra == "all"
-
pytest-benchmark>=4[all-cpu]extra == "all-cpu"
-
pytest-codspeed>=4.4.0[all-macos]extra == "all-macos"
-
pytest-codspeed>=4.4.0[performance]extra == "performance"
-
pytest-codspeed>=4.4.0[all]extra == "all"
-
pytest-codspeed>=4.4.0[all-cpu]extra == "all-cpu"
-
pytest-cov>=6.1.1[all-macos]extra == "all-macos"
-
pytest-cov>=6.1.1[cuda-dev]extra == "cuda-dev"
-
pytest-cov>=6.1.1[dev]extra == "dev"
-
pytest-cov>=6.1.1[test]extra == "test"
-
pytest-cov>=6.1.1[all]extra == "all"
-
pytest-cov>=6.1.1[all-cpu]extra == "all-cpu"
-
pytest-env>=1.0.1[all-cpu]extra == "all-cpu"
-
pytest-env>=1.0.1[all]extra == "all"
-
pytest-env>=1.0.1[all-macos]extra == "all-macos"
-
pytest-env>=1.0.1[cuda-dev]extra == "cuda-dev"
-
pytest-env>=1.0.1[dev]extra == "dev"
-
pytest-env>=1.0.1[test]extra == "test"
-
pytest-json-report>=1.5.0[dev]extra == "dev"
-
pytest-json-report>=1.5.0[test]extra == "test"
-
pytest-json-report>=1.5.0[all]extra == "all"
-
pytest-json-report>=1.5.0[all-cpu]extra == "all-cpu"
-
pytest-json-report>=1.5.0[all-macos]extra == "all-macos"
-
pytest-json-report>=1.5.0[cuda-dev]extra == "cuda-dev"
-
pytest-randomly>=3.15[all-macos]extra == "all-macos"
-
pytest-randomly>=3.15[cuda-dev]extra == "cuda-dev"
-
pytest-randomly>=3.15[dev]extra == "dev"
-
pytest-randomly>=3.15[all]extra == "all"
-
pytest-randomly>=3.15[all-cpu]extra == "all-cpu"
-
pytest-timeout>=2.1[all-macos]extra == "all-macos"
-
pytest-timeout>=2.1[cuda-dev]extra == "cuda-dev"
-
pytest-timeout>=2.1[dev]extra == "dev"
-
pytest-timeout>=2.1[test]extra == "test"
-
pytest-timeout>=2.1[all]extra == "all"
-
pytest-timeout>=2.1[all-cpu]extra == "all-cpu"
-
pytest-xdist>=3.6[all-macos]extra == "all-macos"
-
pytest-xdist>=3.6[cuda-dev]extra == "cuda-dev"
-
pytest-xdist>=3.6[dev]extra == "dev"
-
pytest-xdist>=3.6[test]extra == "test"
-
pytest-xdist>=3.6[all]extra == "all"
-
pytest-xdist>=3.6[all-cpu]extra == "all-cpu"
-
radon>=6.0[all-macos]extra == "all-macos"
-
radon>=6.0[cuda-dev]extra == "cuda-dev"
-
radon>=6.0[dev]extra == "dev"
-
radon>=6.0[all]extra == "all"
-
radon>=6.0[all-cpu]extra == "all-cpu"
-
ruff>=0.1.5[all-macos]extra == "all-macos"
-
ruff>=0.1.5[dev]extra == "dev"
-
ruff>=0.1.5[cuda-dev]extra == "cuda-dev"
-
ruff>=0.1.5[all]extra == "all"
-
ruff>=0.1.5[all-cpu]extra == "all-cpu"
-
ruptures>=1.1[all]extra == "all"
-
ruptures>=1.1[all-cpu]extra == "all-cpu"
-
ruptures>=1.1[all-macos]extra == "all-macos"
-
ruptures>=1.1[changepoint]extra == "changepoint"
-
ruptures>=1.1[test]extra == "test"
-
scikit-learn>=1.4[test]extra == "test"
-
scikit-learn>=1.4[all-macos]extra == "all-macos"
-
scikit-learn>=1.4[all]extra == "all"
-
scikit-learn>=1.4[all-cpu]extra == "all-cpu"
-
scipy-stubs<1.18,>=1.17[dev]extra == "dev"
-
scipy-stubs<1.18,>=1.17[all-macos]extra == "all-macos"
-
scipy-stubs<1.18,>=1.17[all]extra == "all"
-
scipy-stubs<1.18,>=1.17[all-cpu]extra == "all-cpu"
-
scipy-stubs<1.18,>=1.17[cuda-dev]extra == "cuda-dev"
-
substrax>=0.1.15[dev]extra == "dev"
-
substrax>=0.1.15[all]extra == "all"
-
substrax>=0.1.15[all-cpu]extra == "all-cpu"
-
substrax>=0.1.15[all-macos]extra == "all-macos"
-
substrax>=0.1.15[cuda-dev]extra == "cuda-dev"
-
twine>=4.0.2[all]extra == "all"
-
twine>=4.0.2[all-cpu]extra == "all-cpu"
-
twine>=4.0.2[all-macos]extra == "all-macos"
-
twine>=4.0.2[cuda-dev]extra == "cuda-dev"
-
twine>=4.0.2[dev]extra == "dev"
-
wandb>=0.18[all]extra == "all"
-
wandb>=0.18[all-cpu]extra == "all-cpu"
-
wandb>=0.18[all-macos]extra == "all-macos"
-
wandb>=0.18[test]extra == "test"
-
wandb>=0.18[wandb]extra == "wandb"