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
- flax
- gymnax
- jax >=0.4.25
- jaxtyping >=0.2.28
- matplotlib
- numpy >=1.26
- pillow
- safetensors
14 optional dependencies
-
brax>=0.10.3[mabrax]extra == "mabrax"
-
distrax[algs]extra == "algs"
-
flashbax>=0.1.0[algs]extra == "algs"
-
hydra-core>=1.3.2[algs]extra == "algs"
-
imageio[algs]extra == "algs"
-
mpe2[dev]extra == "dev"
-
omegaconf>=2.3.0[algs]extra == "algs"
-
optax[algs]extra == "algs"
-
pre-commit[dev]extra == "dev"
-
pyright[dev]extra == "dev"
-
pytest[dev]extra == "dev"
-
ruff[dev]extra == "dev"
-
tqdm>=4.66.0[algs]extra == "algs"
-
wandb[algs]extra == "algs"