
jax
by jax-ml (organization)
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more (description from the project’s GitHub page)
Repository facts
- Stars
- 36,463
- Forks
- 3,867
- Open issues and pull requests
- 2,571
- Primary language
- Python
- License
- Apache License 2.0 (Apache-2.0)
- Owner
- jax-ml · Organization
- Created
- Oct 25, 2018
- Last push
- Oct 18, 2026
Topics
Add this badge to your README
Shows this repository’s rank in AI on GitHub and links back to this page. Ranks are by stars and change when the list is refreshed. The badge is not an endorsement or a rating. About the badges
Markdown (for a README)
[](https://unitedstatesofamericasuperintelligence.com/github/jax-ml--jax/)
HTML
<a href="https://unitedstatesofamericasuperintelligence.com/github/jax-ml--jax/"><img src="https://unitedstatesofamericasuperintelligence.com/badges/github/jax-ml--jax.svg" alt="Ranked #194 in AI on GitHub" height="20"></a>
Related projects
Projects that share this one’s GitHub topics.
Rank 544: kyegomez/OpenMythos
A theoretical reconstruction of the Claude Mythos architecture, built from first principles using the available research literature.
14,911 starsPythonMITLast push May 23, 2026
Rank 576: unifyai/ivy
Convert Machine Learning Code Between Frameworks
14,167 starsPythonOther licenseLast push Oct 16, 2026
Rank 708: wandb/wandb
The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.
11,301 starsPythonMITLast push Oct 17, 2026
Rank 849: arogozhnikov/einops
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
9,623 starsPythonMITLast push Aug 26, 2026
Rank 1,078: HIPS/autograd
Efficiently computes derivatives of NumPy code.
7,543 starsPythonMITLast push Oct 12, 2026