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
- jax >=0.4.25
- jaxlib
- typing-extensions
3 optional dependencies
-
flax[testing]extra == "testing"
-
pytest[testing]extra == "testing"
-
pytest-cov[testing]extra == "testing"
Used By
- jaxlie 79.3K/mo
- jaxsim 10.1K/mo
- almond-pyroki 1.3K/mo
- almond-jaxls 842/mo
- deepqmc 510/mo
- temgym-core 297/mo
- phlash 188/mo
- ai-helpers-jax-utils 172/mo
- exciting-environments 151/mo
- optispec 79/mo