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
- absl-py >=1.4.0
- beartype >=0.21.0
- chex >=0.1.0
- clu >=0.0.12
- datasets >=4.5.0
- einops >=0.8.0
- etils >=1.10.0
- flax >=0.12.7
- google-metrax >=0.0.1
- grain >=0.2.16
- ir-measures >=0.4.3
-
jax
>=0.10
sys_platform == "darwin"
- jaxtyping >=0.3.7
- keras <3.15,>=3.12.1
- keras-hub <0.31,>=0.25.1
- nvidia-ml-py >=13.590.48
- optax >=0.2.8
- orbax-checkpoint >=0.11.19
- orjson >=3.11.7
- safetensors >=0.4.0
- tokenizers >=0.15.0
- toolz >=1.0.0
- trackio >=0.2.0
1 optional dependencies
-
datatrove>=0.3.0[preprocessing]extra == "preprocessing"