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
- No releases in 385 days despite active commits
((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
- dataclasses-json
- etils
- jax >=0.4.34
- jaxlib
- jaxtyping
- numpy
- orbax-checkpoint >=0.9.0
- protobuf
5 optional dependencies
-
chex[testing]extra == "testing"
-
pytest[testing]extra == "testing"
-
pytest-xdist[testing]extra == "testing"
-
requests[testing]extra == "testing"
-
tf-nightly[testing]extra == "testing"
Used By
- jax2onnx 20.9K/mo
- jax-ai-stack 617/mo
- sequence-layers 118/mo