Rank #194 36,463 starsPython

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)

View on GitHub Homepage

In the USASI catalog: JAX

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

Rank #194 of 9,507 in AI on GitHub, by stars. Data from the GitHub API, fetched Oct 18, 2026; star counts change daily. Stars measure popularity, not quality or safety, and a listing is not an endorsement. How this list is made

Projects that share this one’s GitHub topics.

Rank 81: keras-team/keras
Deep Learning for humans
64,375 starsPythonApache-2.0Last push Oct 16, 2026
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

Repository facts

Topics

Related projects

Rank 81: keras-team/keras

Rank 544: kyegomez/OpenMythos

Rank 576: unifyai/ivy

Rank 708: wandb/wandb

Rank 849: arogozhnikov/einops

Rank 1,078: HIPS/autograd

Support Us