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
- grain >=0.2.18
- jax <0.11.2,>=0.10.2
- optax >=0.2.8
- orbax-checkpoint >=0.12.1
- qwix >=0.1.8
9 optional dependencies
-
jax[tpu]extra == "tpu"
-
jax[cuda12]extra == "cuda12"
-
jax[cuda]extra == "cuda"
-
jax[cuda13]extra == "cuda13"
-
jax[rocm7-local]extra == "rocm7-local"
-
tensorboardx[reporting]extra == "reporting"
-
tensorboardx[tensorboard]extra == "tensorboard"
-
wandb[reporting]extra == "reporting"
-
wandb[wandb]extra == "wandb"