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
- array-record >=0.4
- calibrax >=0.1.14
- flax >=0.12.9
- grain >=0.2.18
- jax >=0.11.1
- jaxtyping >=0.2.20
- numpy >=2.1
- optax >=0.2.8
- orbax-checkpoint >=0.11.33
- protobuf <6,>=5.28.0
- psutil >=5.9.0
- pyyaml >=6
- substrax >=0.1.18
- tomli-w >=1
329 optional dependencies
-
bandit>=1.8.0[dev]extra == "dev"
-
bandit>=1.8.0[cuda-dev]extra == "cuda-dev"
-
bandit>=1.8.0[all-macos]extra == "all-macos"
-
bandit>=1.8.0[all-cpu]extra == "all-cpu"
-
bandit>=1.8.0[all]extra == "all"
-
beartype>=0.14.1[all-cpu]extra == "all-cpu"
-
beartype>=0.14.1[all]extra == "all"
-
beartype>=0.14.1[all-macos]extra == "all-macos"
-
beartype>=0.14.1[test]extra == "test"
-
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[dev]extra == "dev"
-
build>=1.0.3[cuda-dev]extra == "cuda-dev"
-
calibrax>=0.1.14[all]extra == "all"
-
calibrax>=0.1.14[all-macos]extra == "all-macos"
-
calibrax>=0.1.14[wandb]extra == "wandb"
-
calibrax>=0.1.14[all-cpu]extra == "all-cpu"
-
click<8.3[automation]extra == "automation"
-
click>=8.1[benchmark]extra == "benchmark"
-
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"
-
coverage>=7[all]extra == "all"
-
datasets>=2.21[data]extra == "data"
-
datasets>=2.21[benchmark]extra == "benchmark"
-
datasets>=2.21[all-macos]extra == "all-macos"
-
datasets>=2.21[all-cpu]extra == "all-cpu"
-
datasets>=2.21[all]extra == "all"
-
deadcode>=2.4.1[all]extra == "all"
-
deadcode>=2.4.1[dev]extra == "dev"
-
deadcode>=2.4.1[cuda-dev]extra == "cuda-dev"
-
deadcode>=2.4.1[all-macos]extra == "all-macos"
-
deadcode>=2.4.1[all-cpu]extra == "all-cpu"
-
deeplake>=4.0[benchmark]extra == "benchmark"
-
deptry>=0.23.0[all]extra == "all"
-
deptry>=0.23.0[all-macos]extra == "all-macos"
-
deptry>=0.23.0[cuda-dev]extra == "cuda-dev"
-
deptry>=0.23.0[dev]extra == "dev"
-
deptry>=0.23.0[all-cpu]extra == "all-cpu"
-
flake8>=7.0[cuda-dev]extra == "cuda-dev"
-
flake8>=7.0[all]extra == "all"
-
flake8>=7.0[all-cpu]extra == "all-cpu"
-
flake8>=7.0[all-macos]extra == "all-macos"
-
flake8>=7.0[dev]extra == "dev"
-
flake8-functions-names>=0.4.0[dev]extra == "dev"
-
flake8-functions-names>=0.4.0[all]extra == "all"
-
flake8-functions-names>=0.4.0[all-cpu]extra == "all-cpu"
-
flake8-functions-names>=0.4.0[all-macos]extra == "all-macos"
-
flake8-functions-names>=0.4.0[cuda-dev]extra == "cuda-dev"
-
google-cloud-aiplatform>=1.38.0[all]extra == "all"
-
google-cloud-aiplatform>=1.38.0[all-cpu]extra == "all-cpu"
-
google-cloud-aiplatform>=1.38.0[all-macos]extra == "all-macos"
-
google-cloud-aiplatform>=1.38.0[cuda-dev]extra == "cuda-dev"
-
google-cloud-aiplatform>=1.38.0[dev]extra == "dev"
-
google-cloud-storage>=2.14.0[all-cpu]extra == "all-cpu"
-
google-cloud-storage>=2.14.0[all-macos]extra == "all-macos"
-
google-cloud-storage>=2.14.0[cuda-dev]extra == "cuda-dev"
-
google-cloud-storage>=2.14.0[dev]extra == "dev"
-
google-cloud-storage>=2.14.0[all]extra == "all"
-
griffe>=1.7.3[all]extra == "all"
-
griffe>=1.7.3[all-cpu]extra == "all-cpu"
-
griffe>=1.7.3[all-macos]extra == "all-macos"
-
griffe>=1.7.3[docs]extra == "docs"
-
h5py>=3.0[examples]extra == "examples"
-
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[test]extra == "test"
-
hypothesis>=6.100[all-macos]extra == "all-macos"
-
hypothesis>=6.100[all]extra == "all"
-
import-linter>=2.0[all-macos]extra == "all-macos"
-
import-linter>=2.0[all-cpu]extra == "all-cpu"
-
import-linter>=2.0[all]extra == "all"
-
import-linter>=2.0[cuda-dev]extra == "cuda-dev"
-
import-linter>=2.0[dev]extra == "dev"
-
importlib-resources>=5.0[data]extra == "data"
-
importlib-resources>=5.0[all]extra == "all"
-
importlib-resources>=5.0[all-cpu]extra == "all-cpu"
-
importlib-resources>=5.0[all-macos]extra == "all-macos"
-
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]extra == "all"
-
interrogate>=1.5[all-cpu]extra == "all-cpu"
-
ipykernel>=6.29.5[all-macos]extra == "all-macos"
-
ipykernel>=6.29.5[cuda-dev]extra == "cuda-dev"
-
ipykernel>=6.29.5[dev]extra == "dev"
-
ipykernel>=6.29.5[all]extra == "all"
-
ipykernel>=6.29.5[all-cpu]extra == "all-cpu"
-
jax>=0.11.1[cuda12]sys_platform == "linux" and extra == "cuda12"
-
jax>=0.11.1[cuda-dev]sys_platform == "linux" and extra == "cuda-dev"
-
jax>=0.11.1[all]sys_platform == "linux" and extra == "all"
-
jax-dataloader>=0.1[benchmark]extra == "benchmark"
-
jax-metal>=0.1.0[all-macos]sys_platform == "darwin" and extra == "all-macos"
-
jax-metal>=0.1.0[metal]sys_platform == "darwin" and extra == "metal"
-
jupytext>=1.16.0[all-cpu]extra == "all-cpu"
-
jupytext>=1.16.0[all]extra == "all"
-
jupytext>=1.16.0[all-macos]extra == "all-macos"
-
jupytext>=1.16.0[cuda-dev]extra == "cuda-dev"
-
jupytext>=1.16.0[dev]extra == "dev"
-
jupytext>=1.16.0[docs]extra == "docs"
-
keras>=3.0[examples]extra == "examples"
-
librosa>=0.10.0[data]extra == "data"
-
librosa>=0.10.0[all]extra == "all"
-
librosa>=0.10.0[all-cpu]extra == "all-cpu"
-
librosa>=0.10.0[all-macos]extra == "all-macos"
-
litdata>=0.2[benchmark]extra == "benchmark"
-
matplotlib>=3.8[all-macos]extra == "all-macos"
-
matplotlib>=3.8[benchmark]extra == "benchmark"
-
matplotlib>=3.8[examples]extra == "examples"
-
matplotlib>=3.8[all]extra == "all"
-
matplotlib>=3.8[all-cpu]extra == "all-cpu"
-
matplotlib>=3.8[automation]extra == "automation"
-
matplotlib>=3.8[test]extra == "test"
-
mkdocs>=1.6.1[all-cpu]extra == "all-cpu"
-
mkdocs>=1.6.1[all]extra == "all"
-
mkdocs>=1.6.1[all-macos]extra == "all-macos"
-
mkdocs>=1.6.1[docs]extra == "docs"
-
mkdocs-gen-files>=0.5.0[docs]extra == "docs"
-
mkdocs-gen-files>=0.5.0[all]extra == "all"
-
mkdocs-gen-files>=0.5.0[all-cpu]extra == "all-cpu"
-
mkdocs-gen-files>=0.5.0[all-macos]extra == "all-macos"
-
mkdocs-include-exclude-files>=0.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[docs]extra == "docs"
-
mkdocs-ipynb>=0.1[all]extra == "all"
-
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-jupyter>=0.24[docs]extra == "docs"
-
mkdocs-jupyter>=0.24[all]extra == "all"
-
mkdocs-jupyter>=0.24[all-cpu]extra == "all-cpu"
-
mkdocs-jupyter>=0.24[all-macos]extra == "all-macos"
-
mkdocs-literate-nav>=0.6.0[docs]extra == "docs"
-
mkdocs-literate-nav>=0.6.0[all-macos]extra == "all-macos"
-
mkdocs-literate-nav>=0.6.0[all-cpu]extra == "all-cpu"
-
mkdocs-literate-nav>=0.6.0[all]extra == "all"
-
mkdocs-material>=9.6.7[all-macos]extra == "all-macos"
-
mkdocs-material>=9.6.7[all-cpu]extra == "all-cpu"
-
mkdocs-material>=9.6.7[all]extra == "all"
-
mkdocs-material>=9.6.7[docs]extra == "docs"
-
mkdocstrings>=0.28.3[docs]extra == "docs"
-
mkdocstrings>=0.28.3[all-cpu]extra == "all-cpu"
-
mkdocstrings>=0.28.3[all]extra == "all"
-
mkdocstrings>=0.28.3[all-macos]extra == "all-macos"
-
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"
-
mkdocstrings-python>=1.1.2[docs]extra == "docs"
-
mosaicml-streaming>=0.7[benchmark]extra == "benchmark"
-
nvidia-dali-cuda120>=1.0[benchmark]sys_platform == "linux" and extra == "benchmark"
-
pillow>=10.0.0[data]extra == "data"
-
pillow>=10.0.0[all-macos]extra == "all-macos"
-
pillow>=10.0.0[all-cpu]extra == "all-cpu"
-
pillow>=10.0.0[all]extra == "all"
-
pip-audit>=2.7[all-cpu]extra == "all-cpu"
-
pip-audit>=2.7[all]extra == "all"
-
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"
-
pre-commit>=4.3.0[cuda-dev]extra == "cuda-dev"
-
pre-commit>=4.3.0[dev]extra == "dev"
-
pre-commit>=4.3.0[all]extra == "all"
-
pre-commit>=4.3.0[all-cpu]extra == "all-cpu"
-
pre-commit>=4.3.0[all-macos]extra == "all-macos"
-
pyarrow>=21[all-cpu]extra == "all-cpu"
-
pyarrow>=21[all-macos]extra == "all-macos"
-
pyarrow>=21[all]extra == "all"
-
pyarrow>=21[data]extra == "data"
-
pyarrow>=21[benchmark]extra == "benchmark"
-
pydeps>=1.12.17[cuda-dev]extra == "cuda-dev"
-
pydeps>=1.12.17[dev]extra == "dev"
-
pydeps>=1.12.17[all-cpu]extra == "all-cpu"
-
pydeps>=1.12.17[all]extra == "all"
-
pydeps>=1.12.17[all-macos]extra == "all-macos"
-
pylint>=3.0[cuda-dev]extra == "cuda-dev"
-
pylint>=3.0[dev]extra == "dev"
-
pylint>=3.0[all]extra == "all"
-
pylint>=3.0[all-cpu]extra == "all-cpu"
-
pylint>=3.0[all-macos]extra == "all-macos"
-
pymdown-extensions>=10.14.3[docs]extra == "docs"
-
pymdown-extensions>=10.14.3[all]extra == "all"
-
pymdown-extensions>=10.14.3[all-cpu]extra == "all-cpu"
-
pymdown-extensions>=10.14.3[all-macos]extra == "all-macos"
-
pyright>=1.1.336[cuda-dev]extra == "cuda-dev"
-
pyright>=1.1.336[dev]extra == "dev"
-
pyright>=1.1.336[all-macos]extra == "all-macos"
-
pyright>=1.1.336[all-cpu]extra == "all-cpu"
-
pyright>=1.1.336[all]extra == "all"
-
pytest>=8.3.5[cuda-dev]extra == "cuda-dev"
-
pytest>=8.3.5[dev]extra == "dev"
-
pytest>=8.3.5[test]extra == "test"
-
pytest>=8.3.5[all]extra == "all"
-
pytest>=8.3.5[all-cpu]extra == "all-cpu"
-
pytest>=8.3.5[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-asyncio>=0.23[all-macos]extra == "all-macos"
-
pytest-benchmark>=4[all-macos]extra == "all-macos"
-
pytest-benchmark>=4[all]extra == "all"
-
pytest-benchmark>=4[all-cpu]extra == "all-cpu"
-
pytest-benchmark>=4[cuda-dev]extra == "cuda-dev"
-
pytest-benchmark>=4[dev]extra == "dev"
-
pytest-benchmark>=4[test]extra == "test"
-
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-macos]extra == "all-macos"
-
pytest-cov>=6.1.1[all-cpu]extra == "all-cpu"
-
pytest-cov>=6.1.1[all]extra == "all"
-
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-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-json-report>=1.5.0[cuda-dev]extra == "cuda-dev"
-
pytest-json-report>=1.5.0[dev]extra == "dev"
-
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-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-randomly>=3.15[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-timeout>=2.1[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"
-
pytest-xdist>=3.6[all-macos]extra == "all-macos"
-
python-dotenv>=1[cuda-dev]extra == "cuda-dev"
-
python-dotenv>=1[all]extra == "all"
-
python-dotenv>=1[all-cpu]extra == "all-cpu"
-
python-dotenv>=1[all-macos]extra == "all-macos"
-
python-dotenv>=1[dev]extra == "dev"
-
radon>=6.0[all]extra == "all"
-
radon>=6.0[cuda-dev]extra == "cuda-dev"
-
radon>=6.0[dev]extra == "dev"
-
radon>=6.0[all-macos]extra == "all-macos"
-
radon>=6.0[all-cpu]extra == "all-cpu"
-
ray>=2.9[benchmark]extra == "benchmark"
-
rich>=13.0[benchmark]extra == "benchmark"
-
ruff>=0.1.5[dev]extra == "dev"
-
ruff>=0.1.5[all]extra == "all"
-
ruff>=0.1.5[all-cpu]extra == "all-cpu"
-
ruff>=0.1.5[all-macos]extra == "all-macos"
-
ruff>=0.1.5[cuda-dev]extra == "cuda-dev"
-
shellcheck-py>=0.10.0.1[cuda-dev]extra == "cuda-dev"
-
shellcheck-py>=0.10.0.1[all-macos]extra == "all-macos"
-
shellcheck-py>=0.10.0.1[all-cpu]extra == "all-cpu"
-
shellcheck-py>=0.10.0.1[all]extra == "all"
-
shellcheck-py>=0.10.0.1[dev]extra == "dev"
-
skypilot<0.12,>=0.11[automation]extra == "automation"
-
soundfile>=0.12.1[all-macos]extra == "all-macos"
-
soundfile>=0.12.1[data]extra == "data"
-
soundfile>=0.12.1[all-cpu]extra == "all-cpu"
-
soundfile>=0.12.1[all]extra == "all"
-
spdl>=0.1[all-macos]sys_platform == "linux" and extra == "all-macos"
-
spdl>=0.1[benchmark]extra == "benchmark"
-
spdl>=0.1[test]sys_platform == "linux" and extra == "test"
-
spdl>=0.1[all]sys_platform == "linux" and extra == "all"
-
spdl>=0.1[all-cpu]sys_platform == "linux" and extra == "all-cpu"
-
substrax>=0.1.18[test]extra == "test"
-
substrax>=0.1.18[all]extra == "all"
-
substrax>=0.1.18[all-cpu]extra == "all-cpu"
-
substrax>=0.1.18[all-macos]extra == "all-macos"
-
tabulate>=0.9[benchmark]extra == "benchmark"
-
tensorflow<2.21,>=2.20.0[tfds]extra == "tfds"
-
tensorflow-datasets>=4.9.2[all-cpu]extra == "all-cpu"
-
tensorflow-datasets>=4.9.2[all-macos]extra == "all-macos"
-
tensorflow-datasets>=4.9[benchmark]extra == "benchmark"
-
tensorflow-datasets>=4.9.2[data]extra == "data"
-
tensorflow-datasets>=4.9.2[tfds]extra == "tfds"
-
tensorflow-datasets>=4.9.2[all]extra == "all"
-
torch<2.11,>=2.0[all]extra == "all"
-
torch<2.11,>=2.0[all-cpu]extra == "all-cpu"
-
torch<2.11,>=2.0[all-macos]extra == "all-macos"
-
torch<2.11,>=2.0[benchmark]extra == "benchmark"
-
torch<2.11,>=2.0[data]extra == "data"
-
torchaudio<2.11,>=2.0[all]extra == "all"
-
torchaudio<2.11,>=2.0[all-cpu]extra == "all-cpu"
-
torchaudio<2.11,>=2.0[all-macos]extra == "all-macos"
-
torchaudio<2.11,>=2.0[benchmark]extra == "benchmark"
-
torchaudio<2.11,>=2.0[data]extra == "data"
-
torchcrepe>=0.0.24[all]extra == "all"
-
torchcrepe>=0.0.24[all-cpu]extra == "all-cpu"
-
torchcrepe>=0.0.24[all-macos]extra == "all-macos"
-
torchcrepe>=0.0.24[data]extra == "data"
-
torchcrepe[examples]extra == "examples"
-
torchvision<0.26,>=0.15[benchmark]extra == "benchmark"
-
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"
-
vastai-sdk<0.3,>=0.2.6[automation]extra == "automation"
-
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"
-
webdataset>=0.2[benchmark]extra == "benchmark"
-
wemake-python-styleguide>=1.0[all]extra == "all"
-
wemake-python-styleguide>=1.0[all-cpu]extra == "all-cpu"
-
wemake-python-styleguide>=1.0[all-macos]extra == "all-macos"
-
wemake-python-styleguide>=1.0[cuda-dev]extra == "cuda-dev"
-
wemake-python-styleguide>=1.0[dev]extra == "dev"
-
xenon>=0.9.3[all]extra == "all"
-
xenon>=0.9.3[all-cpu]extra == "all-cpu"
-
xenon>=0.9.3[all-macos]extra == "all-macos"
-
xenon>=0.9.3[cuda-dev]extra == "cuda-dev"
-
xenon>=0.9.3[dev]extra == "dev"