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
- flatbuffers >=24
- huggingface-hub <3,>=2.2
- jax ==0.10.1
- numpy
- pydantic
- scipy
35 optional dependencies
-
flashbax==0.1.3[training]extra == "training"
-
flashbax==0.1.3[all]extra == "all"
-
flax[all]extra == "all"
-
flax[training]extra == "training"
-
hydra-core==1.3.7[hydra]extra == "hydra"
-
hydra-core==1.3.7[all]extra == "all"
-
hydra-core==1.3.7[dev]extra == "dev"
-
ipykernel[all]extra == "all"
-
ipykernel[notebook]extra == "notebook"
-
ipywidgets[notebook]extra == "notebook"
-
ipywidgets[all]extra == "all"
-
jax==0.10.1[rocm](sys_platform == "linux" and platform_machine == "x86_64") and extra == "rocm"
-
jax==0.10.1[cuda13]extra == "cuda13"
-
jax==0.10.1[cuda12]extra == "cuda12"
-
jax==0.10.1[tpu](sys_platform == "linux" and platform_machine == "x86_64") and extra == "tpu"
-
jax-rocm7-plugin==0.10.1[rocm](sys_platform == "linux" and platform_machine == "x86_64") and extra == "rocm"
-
matplotlib[training]extra == "training"
-
matplotlib[all]extra == "all"
-
matplotlib[viz]extra == "viz"
-
nbconvert[notebook]extra == "notebook"
-
nbconvert[all]extra == "all"
-
omegaconf==2.3.0[all]extra == "all"
-
omegaconf==2.3.0[hydra]extra == "hydra"
-
omegaconf==2.3.0[dev]extra == "dev"
-
optax[all]extra == "all"
-
optax[training]extra == "training"
-
orbax-checkpoint[all]extra == "all"
-
orbax-checkpoint[training]extra == "training"
-
pandas[notebook]extra == "notebook"
-
pandas[all]extra == "all"
-
pre-commit[dev]extra == "dev"
-
pyright[dev]extra == "dev"
-
pytest[dev]extra == "dev"
-
ruff[dev]extra == "dev"
-
wandb[wandb]extra == "wandb"