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 >=0.1.90
- flax ==0.10.3
- imageio >=2.37.0
- jax ==0.4.38
- jaxlib ==0.4.38
- matplotlib >=3.8.0
- numpy >=1.26.0
- pillow >=10.0.0
- scipy >=1.12.0
- seaborn >=0.13.0
26 optional dependencies
-
anthropic>=0.40.0[baselines-llm]extra == "baselines-llm"
-
beartype>=0.18.0[dev]extra == "dev"
-
distrax>=0.1.5[baselines-rl]extra == "baselines-rl"
-
google-genai>=0.3.0[baselines-llm]extra == "baselines-llm"
-
hydra-core>=1.3.0[baselines-llm]extra == "baselines-llm"
-
hydra-core>=1.3.0[baselines-rl]extra == "baselines-rl"
-
jax==0.4.38[gpu]extra == "gpu"
-
jax==0.4.38[cuda12]extra == "cuda12"
-
jaxmarl>=0.0.3[baselines-rl]extra == "baselines-rl"
-
jaxtyping>=0.2.34[dev]extra == "dev"
-
omegaconf>=2.3.0[baselines-rl]extra == "baselines-rl"
-
omegaconf>=2.3.0[baselines-llm]extra == "baselines-llm"
-
openai>=1.0.0[llm]extra == "llm"
-
openai>=1.0.0[baselines-llm]extra == "baselines-llm"
-
optax>=0.2.0[baselines-rl]extra == "baselines-rl"
-
orbax-checkpoint>=0.5.0[baselines-rl]extra == "baselines-rl"
-
pandas<3.0[baselines-llm]extra == "baselines-llm"
-
pygame>=2.6.0[play]extra == "play"
-
pytest>=8.0.0[dev]extra == "dev"
-
pytest-cov>=5.0.0[dev]extra == "dev"
-
pyyaml>=6.0[baselines-rl]extra == "baselines-rl"
-
rliable>=1.2.0[baselines-llm]extra == "baselines-llm"
-
ruff>=0.4.0[dev]extra == "dev"
-
tqdm>=4.66.0[baselines-llm]extra == "baselines-llm"
-
wandb>=0.16.0[baselines-llm]extra == "baselines-llm"
-
wandb>=0.16.0[baselines-rl]extra == "baselines-rl"