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
- etils >=1.14.0
- flax >=0.12.9
- jax <0.11.2,>=0.11.1
- jaxlib <0.11.2,>=0.11.1
- numpy <2.6.0,>=1.24
- optax >=0.2.8
- orbax-checkpoint >=0.11.33
- pydantic >=2.10
185 optional dependencies
-
bandit>=1.8.6[all-cpu]extra == "all-cpu"
-
bandit>=1.8.6[cuda-dev]extra == "cuda-dev"
-
bandit>=1.8.6[all-macos]extra == "all-macos"
-
bandit>=1.8.6[all]extra == "all"
-
bandit>=1.8.6[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"
-
build>=1.0.3[cuda-dev]extra == "cuda-dev"
-
build>=1.0.3[dev]extra == "dev"
-
coverage>=7[cuda-dev]extra == "cuda-dev"
-
coverage>=7[all]extra == "all"
-
coverage>=7[all-cpu]extra == "all-cpu"
-
coverage>=7[dev]extra == "dev"
-
coverage>=7[test]extra == "test"
-
coverage>=7[all-macos]extra == "all-macos"
-
deptry>=0.23[all-macos]extra == "all-macos"
-
deptry>=0.23[cuda-dev]extra == "cuda-dev"
-
deptry>=0.23[dev]extra == "dev"
-
deptry>=0.23[all-cpu]extra == "all-cpu"
-
deptry>=0.23[all]extra == "all"
-
diff-cover>=10.6.0[all-macos]extra == "all-macos"
-
diff-cover>=10.6.0[test]extra == "test"
-
diff-cover>=10.6.0[all]extra == "all"
-
diff-cover>=10.6.0[all-cpu]extra == "all-cpu"
-
griffe>=1.7.3[all-cpu]extra == "all-cpu"
-
griffe>=1.7.3[docs]extra == "docs"
-
griffe>=1.7.3[all-macos]extra == "all-macos"
-
griffe>=1.7.3[all]extra == "all"
-
hypothesis>=6.100[test]extra == "test"
-
hypothesis>=6.100[all-macos]extra == "all-macos"
-
hypothesis>=6.100[all-cpu]extra == "all-cpu"
-
hypothesis>=6.100[all]extra == "all"
-
import-linter>=2.0[all-cpu]extra == "all-cpu"
-
import-linter>=2.0[all]extra == "all"
-
import-linter>=2.0[all-macos]extra == "all-macos"
-
import-linter>=2.0[cuda-dev]extra == "cuda-dev"
-
import-linter>=2.0[dev]extra == "dev"
-
interrogate>=1.5[dev]extra == "dev"
-
interrogate>=1.5[all-macos]extra == "all-macos"
-
interrogate>=1.5[cuda-dev]extra == "cuda-dev"
-
interrogate>=1.5[all]extra == "all"
-
interrogate>=1.5[all-cpu]extra == "all-cpu"
-
jax<0.11.2,>=0.11.1[cuda-dev]extra == "cuda-dev"
-
jax<0.11.2,>=0.11.1[cuda12]extra == "cuda12"
-
jax<0.11.2,>=0.11.1[all]extra == "all"
-
jax-metal>=0.1.0[all-macos]extra == "all-macos"
-
jax-metal>=0.1.0[metal]extra == "metal"
-
matplotlib>=3.8[test]extra == "test"
-
matplotlib>=3.8[all-macos]extra == "all-macos"
-
matplotlib>=3.8[plots]extra == "plots"
-
matplotlib>=3.8[all-cpu]extra == "all-cpu"
-
matplotlib>=3.8[all]extra == "all"
-
mkdocs<2.0,>=1.6.1[docs]extra == "docs"
-
mkdocs<2.0,>=1.6.1[all-macos]extra == "all-macos"
-
mkdocs<2.0,>=1.6.1[all]extra == "all"
-
mkdocs<2.0,>=1.6.1[all-cpu]extra == "all-cpu"
-
mkdocs-material>=9.7.7[all]extra == "all"
-
mkdocs-material>=9.7.7[all-cpu]extra == "all-cpu"
-
mkdocs-material>=9.7.7[all-macos]extra == "all-macos"
-
mkdocs-material>=9.7.7[docs]extra == "docs"
-
mkdocstrings>=0.28.3[docs]extra == "docs"
-
mkdocstrings>=0.28.3[all-macos]extra == "all-macos"
-
mkdocstrings>=0.28.3[all-cpu]extra == "all-cpu"
-
mkdocstrings>=0.28.3[all]extra == "all"
-
mkdocstrings-python>=1.1.2[docs]extra == "docs"
-
mkdocstrings-python>=1.1.2[all]extra == "all"
-
mkdocstrings-python>=1.1.2[all-cpu]extra == "all-cpu"
-
mkdocstrings-python>=1.1.2[all-macos]extra == "all-macos"
-
mlflow>=2[mlflow]extra == "mlflow"
-
mlflow>=2[all-macos]extra == "all-macos"
-
mlflow>=2[all-cpu]extra == "all-cpu"
-
mlflow>=2[all]extra == "all"
-
modal>=1.5.5[modal]extra == "modal"
-
modal>=1.5.5[cuda-dev]extra == "cuda-dev"
-
modal>=1.5.5[all]extra == "all"
-
modal>=1.5.5[all-cpu]extra == "all-cpu"
-
modal>=1.5.5[all-macos]extra == "all-macos"
-
modal>=1.5.5[dev]extra == "dev"
-
pip-audit>=2.7[cuda-dev]extra == "cuda-dev"
-
pip-audit>=2.7[all]extra == "all"
-
pip-audit>=2.7[all-cpu]extra == "all-cpu"
-
pip-audit>=2.7[all-macos]extra == "all-macos"
-
pip-audit>=2.7[dev]extra == "dev"
-
pre-commit>=4.2[all-cpu]extra == "all-cpu"
-
pre-commit>=4.2[all-macos]extra == "all-macos"
-
pre-commit>=4.2[cuda-dev]extra == "cuda-dev"
-
pre-commit>=4.2[dev]extra == "dev"
-
pre-commit>=4.2[all]extra == "all"
-
pydoclint>=0.5[all]extra == "all"
-
pydoclint>=0.5[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"
-
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"
-
pygments<3.0,>=2.20[all-macos]extra == "all-macos"
-
pylint>=3.3[all]extra == "all"
-
pylint>=3.3[all-cpu]extra == "all-cpu"
-
pylint>=3.3[all-macos]extra == "all-macos"
-
pylint>=3.3[cuda-dev]extra == "cuda-dev"
-
pylint>=3.3[dev]extra == "dev"
-
pymdown-extensions>=11.0.1[all]extra == "all"
-
pymdown-extensions>=11.0.1[all-cpu]extra == "all-cpu"
-
pymdown-extensions>=11.0.1[all-macos]extra == "all-macos"
-
pymdown-extensions>=11.0.1[docs]extra == "docs"
-
pyright>=1.1.403[all]extra == "all"
-
pyright>=1.1.403[all-cpu]extra == "all-cpu"
-
pyright>=1.1.403[all-macos]extra == "all-macos"
-
pyright>=1.1.403[cuda-dev]extra == "cuda-dev"
-
pyright>=1.1.403[dev]extra == "dev"
-
pytest>=9.0.3[all]extra == "all"
-
pytest>=9.0.3[all-cpu]extra == "all-cpu"
-
pytest>=9.0.3[all-macos]extra == "all-macos"
-
pytest>=9.0.3[test]extra == "test"
-
pytest>=9.0.3[testing]extra == "testing"
-
pytest-cov>=6.1.1[all]extra == "all"
-
pytest-cov>=6.1.1[all-cpu]extra == "all-cpu"
-
pytest-cov>=6.1.1[all-macos]extra == "all-macos"
-
pytest-cov>=6.1.1[test]extra == "test"
-
pytest-env>=1.0.1[all]extra == "all"
-
pytest-env>=1.0.1[all-cpu]extra == "all-cpu"
-
pytest-env>=1.0.1[all-macos]extra == "all-macos"
-
pytest-env>=1.0.1[test]extra == "test"
-
pytest-json-report>=1.5.0[all-macos]extra == "all-macos"
-
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[test]extra == "test"
-
pytest-randomly>=3.15[all]extra == "all"
-
pytest-randomly>=3.15[all-cpu]extra == "all-cpu"
-
pytest-randomly>=3.15[all-macos]extra == "all-macos"
-
pytest-randomly>=3.15[test]extra == "test"
-
pytest-timeout>=2.1[all]extra == "all"
-
pytest-timeout>=2.1[all-cpu]extra == "all-cpu"
-
pytest-timeout>=2.1[all-macos]extra == "all-macos"
-
pytest-timeout>=2.1[test]extra == "test"
-
pytest-xdist>=3.6[all]extra == "all"
-
pytest-xdist>=3.6[all-cpu]extra == "all-cpu"
-
pytest-xdist>=3.6[all-macos]extra == "all-macos"
-
pytest-xdist>=3.6[test]extra == "test"
-
pyyaml>=6.0[all]extra == "all"
-
pyyaml>=6.0[all-cpu]extra == "all-cpu"
-
pyyaml>=6.0[all-macos]extra == "all-macos"
-
pyyaml>=6.0[cuda-dev]extra == "cuda-dev"
-
pyyaml>=6.0[dev]extra == "dev"
-
radon>=6.0[all]extra == "all"
-
radon>=6.0[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"
-
ruff>=0.15.2[all]extra == "all"
-
ruff>=0.15.2[all-cpu]extra == "all-cpu"
-
ruff>=0.15.2[all-macos]extra == "all-macos"
-
ruff>=0.15.2[cuda-dev]extra == "cuda-dev"
-
ruff>=0.15.2[dev]extra == "dev"
-
toml-sort>=0.23.1[all]extra == "all"
-
toml-sort>=0.23.1[all-cpu]extra == "all-cpu"
-
toml-sort>=0.23.1[all-macos]extra == "all-macos"
-
toml-sort>=0.23.1[cuda-dev]extra == "cuda-dev"
-
toml-sort>=0.23.1[dev]extra == "dev"
-
twine>=6.1[all]extra == "all"
-
twine>=6.1[all-cpu]extra == "all-cpu"
-
twine>=6.1[all-macos]extra == "all-macos"
-
twine>=6.1[cuda-dev]extra == "cuda-dev"
-
twine>=6.1[dev]extra == "dev"
-
validate-pyproject>=0.23[all]extra == "all"
-
validate-pyproject>=0.23[all-cpu]extra == "all-cpu"
-
validate-pyproject>=0.23[all-macos]extra == "all-macos"
-
validate-pyproject>=0.23[cuda-dev]extra == "cuda-dev"
-
validate-pyproject>=0.23[dev]extra == "dev"
-
vulture>=2.14[all]extra == "all"
-
vulture>=2.14[all-cpu]extra == "all-cpu"
-
vulture>=2.14[all-macos]extra == "all-macos"
-
vulture>=2.14[cuda-dev]extra == "cuda-dev"
-
vulture>=2.14[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[wandb]extra == "wandb"
-
xenon>=0.9[all]extra == "all"
-
xenon>=0.9[all-cpu]extra == "all-cpu"
-
xenon>=0.9[all-macos]extra == "all-macos"
-
xenon>=0.9[cuda-dev]extra == "cuda-dev"
-
xenon>=0.9[dev]extra == "dev"