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
- braceexpand >=0.1.7
- chex >=0.1.91
- contextlib2 ~=21.6.0
- fsspec <2026,>=2024.2
- google-cloud-storage >=3.3.0
- jax >=0.9.2
- jaxtyping >=0.3.2
- mergedeep ~=1.3.4
- ml-collections ==1.1.0
- optax >=0.2.2
- plum-dispatch ==2.3.2
- psutil >=7.0.0
- pyyaml ~=6.0.2
- ray >=2.54.0
- safetensors >=0.6.2
- tblib ~=3.0.0
- tensorstore >=0.1.76
- tqdm >=4.67.1
1 optional dependencies
-
pytest[dev]extra == "dev"
Used By
- easydel 1.8K/mo
- jax-flash-attn2 81/mo
- ejgpu 59/mo
- dukron 56/mo