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
- einops ==0.8.1
- flax ==0.12.10
- jax ==0.11.2
- numba ==0.62.0
- numpy ==2.3.3
- optax ==0.2.5
- orbax-checkpoint ==0.11.25
- scipy ==1.16.2
11 optional dependencies
-
jax==0.11.2[cuda13]extra == "cuda13"
-
jax==0.11.2[cuda12]extra == "cuda12"
-
jax==0.11.2[cuda]extra == "cuda"
-
jsonargparse==4.41.0[wandb]extra == "wandb"
-
myst-parser>=2.0[docs]extra == "docs"
-
pytest[test]extra == "test"
-
sphinx>=8.2[docs]extra == "docs"
-
sphinx-book-theme>=1.4[docs]extra == "docs"
-
sphinx-copybutton>=0.5[docs]extra == "docs"
-
sphinx-design>=0.5[docs]extra == "docs"
-
wandb==0.27.0[wandb]extra == "wandb"