Downloads in the 30 and 7 days to October 10, 2026. Refreshed weekly.
Package Information
Latest Version
Requires Python
>=3.12
Licence
Apache-2.0
Last Release
4 weeks ago
Author
JAX team
First Release
July 15, 2025
Total Releases
16
Python Versions
3.12
3.13
3.14
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
Score breakdown - each component scores 0-100, weighted by percentage
Maintenance
90
Popularity
93
Adoption
96
Community
80
Provenance
0
How the maintenance status was determined
Last commit
2 days ago
Monthly downloads
442,227
•
Active
Activity within 30 days
Stable
Activity within 31-180 days
Minimal
Activity within 181-365 days, or no activity data
At Risk
No activity for > 365 days, but > 10,000 monthly downloads
Archived
Repository flagged as archived on GitHub
How the popularity category was determined
Monthly downloads
442,227
Stars
36,417
Stars per 1k downloads
82.35
Industry Standard
> 10,000,000 monthly downloads
Hidden Gem
< 0.5 stars per 1k downloads and > 100,000 monthly downloads
•
Community Favorite
> 5 stars per 1k downloads
Rising Star
Balanced ratio of stars to downloads
Unknown
Fewer than 1,000 monthly downloads, or no star data
How download momentum is calculated
Week daily avg
35,817
Month daily avg
14,741
Momentum
+143.0%
((week_daily_avg - month_daily_avg) / month_daily_avg) * 100More insights and metrics
Python version support tiers
requires_python
>=3.12
Cutting Edge
Minimum supported version is 3.14 or newer
•
Modern
Minimum supported version is 3.11 to 3.13
Legacy
Supports 3.10 or older (or Python 2)
Unknown
No requires_python specified or unparseable
License categories
License
Apache-2.0
•
Permissive
MIT, Apache, BSD, ISC, Unlicense, CC0, etc.
Copyleft
GPL, LGPL, AGPL, MPL, EPL, etc.
Proprietary
Commercial, all rights reserved, custom EULAs
Unknown
No license specified, or none of the above match
How the release cadence was determined
Last release
28 days ago
Total releases
16
Releases/year
12.8
•
Rapid
Last release within 365 days and > 12 releases/year
Regular
Last release within 365 days
Slow
Last release 366-730 days ago
Stalled
Last release > 730 days ago, or no releases
Unknown
No usable release date available
How contributor risk was determined
Contributors
987
Monthly downloads
442,227
High Load
≤ 2 contributors and ≥ 50,000 monthly downloads
Moderate Load
< 10 contributors and ≥ 500,000 monthly downloads
Low Load
< 25 contributors and ≥ 5,000,000 monthly downloads, or sole maintainer with ≥ 10,000 monthly downloads
•
Balanced
Enough contributors for the package's popularity
Unknown
No contributor data available
Engagement
85
Issue Burden
47.2
per 1k stars
Discussions
81.1
per 1k stars
Fork Ratio
25.6%
contrib / forks
Stale Release
No
PR Velocity
Active
Not all downloads include detailed metadata. Breakdowns reflect the subset that does.
Downloads by OS
Linux
72.1K
macOS
4
Windows
1
Python Versions
3.13
97.0%
70.0K
3.12
2.6%
1.9K
3.11
0.3%
200
3.14
0.1%
81
3.15
0.0%
4
Top Countries
1.
US
72.7K
US
2.
SG
49
SG
3.
CN
48
CN
4.
GB
46
GB
5.
FR
42
FR
6.
JP
32
JP
7.
NL
32
NL
8.
CA
31
CA
9.
DE
25
DE
10.
FI
16
FI
Weekly Downloads
GitHub Stars *
Dependencies
1
- jax-cuda13-pjrt ==0.11.2
13 optional dependencies
-
nvidia-cublas>=13.0.0.19[with-cuda]sys_platform == "linux" and extra == "with-cuda"
-
nvidia-cuda-cupti>=13.0.48[with-cuda]sys_platform == "linux" and extra == "with-cuda"
-
nvidia-cuda-nvcc>=13.0.48[with-cuda]sys_platform == "linux" and extra == "with-cuda"
-
nvidia-cuda-nvrtc>=13.0.48[with-cuda]sys_platform == "linux" and extra == "with-cuda"
-
nvidia-cuda-runtime>=13.0.48[with-cuda]sys_platform == "linux" and extra == "with-cuda"
-
nvidia-cudnn-cu13<10.0,>=9.12.0.46[with-cuda]sys_platform == "linux" and extra == "with-cuda"
-
nvidia-cufft>=12.0.0.15[with-cuda]sys_platform == "linux" and extra == "with-cuda"
-
nvidia-cusolver>=12.0.3.29[with-cuda]sys_platform == "linux" and extra == "with-cuda"
-
nvidia-cusparse>=12.6.2.49[with-cuda]sys_platform == "linux" and extra == "with-cuda"
-
nvidia-nccl-cu13>=2.27.7[with-cuda]sys_platform == "linux" and extra == "with-cuda"
-
nvidia-nvjitlink>=13.0.39[with-cuda]sys_platform == "linux" and extra == "with-cuda"
-
nvidia-nvshmem-cu13>=3.3.20[with-cuda]sys_platform == "linux" and extra == "with-cuda"
-
nvidia-nvvm[with-cuda]extra == "with-cuda"
Used By
0
Not used by any tracked packages
Data and sources
Tracking Since
September 26, 2026
Data Updated
October 11, 2026
Data Freshness
Fresh