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
- flashbax
- jax >=0.4.27.0
- jax-tqdm
- jaxlib >=0.4.27.0
- jaxtyping
- matplotlib
- networkx
- numpy ==1.26.4
- omegaconf
11 optional dependencies
-
equinox[tests]extra == "tests"
-
equinox[baselines]extra == "baselines"
-
flashbax[baselines]extra == "baselines"
-
hydra-core[baselines]extra == "baselines"
-
jax-tqdm[baselines]extra == "baselines"
-
jraph[baselines]extra == "baselines"
-
optax[baselines]extra == "baselines"
-
orbax-checkpoint[baselines]extra == "baselines"
-
pillow[baselines]extra == "baselines"
-
pytest[tests]extra == "tests"
-
wandb[baselines]extra == "baselines"