jax-ml
@jax-ml • Pushing back the limits on numerical computing.. Use this route to separate flagship concentration from portfolio breadth before you treat a publisher as broadly strong.
Portfolio concentration
95%
Top three share
Shows whether the organization is driven by one breakout repo or several visible projects.
Breadth
13 repos
Visible snapshot
9 repositories updated in the last 90 days.
Leading language
Python
Portfolio mix
Python (8), C++ (2), HTML (1)
Average size
3K
Stars per repository
Useful for distinguishing one flagship-heavy publisher from a repeatable portfolio.
Updated: 2026-01-10(155d ago)GitHub API fallback13 repositories
95%
of the visible star count comes from this organization's top three repositories.
3K
stars per repository in this same snapshot.
Python
is the most common language here, with 9 repositories updated in the last 90 days.
Why this rank
This organization stands out because one flagship repo drives 91% of its visible star count.
Flagship share 91%Breakout repo: jax
Organization pages work best when you separate portfolio breadth from flagship concentration. In jax-ml's case, the visible top three repositories account for about 95% of total stars in this snapshot, which helps explain whether the organization is known for one breakout project or for a broader repeatable portfolio.
The dominant language mix here is Python (8), C++ (2), HTML (1). That makes this page useful not just for popularity checks, but also for seeing what technical shape an organization's public ecosystem actually has.
Top Repositories
| # | Repository | Language | Stars | 🍴 Forks | Updated |
|---|---|---|---|---|---|
| 1 | jax-ml/jax Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more | Python | 35.8K | 3.6K | Today |
| 2 | jax-ml/scaling-book Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs | HTML | 1.1K | 161 | 3 days ago |
| 3 | jax-ml/jax-triton jax-triton contains integrations between JAX and OpenAI Triton | Python | 462 | 57 | 1 weeks ago |
| 4 | jax-ml/ml_dtypes A stand-alone implementation of several NumPy dtype extensions used in machine learning. | C++ | 347 | 58 | 2 days ago |
| 5 | jax-ml/oryx Oryx is a library for probabilistic programming and deep learning built on top of Jax. | Python | 317 | 13 | 1 weeks ago |
| 6 | jax-ml/jax-ai-stack | Python | 303 | 47 | 5 days ago |
| 7 | jax-ml/jax-llm-examples Minimal yet performant LLM examples in pure JAX | Python | 256 | 35 | 2 months ago |
| 8 | jax-ml/bayeux State of the art inference for your bayesian models. | Python | 242 | 13 | 5 months ago |
| 9 | jax-ml/bonsai Minimal, lightweight JAX implementations of popular models. | Python | 236 | 46 | 2 weeks ago |
| 10 | jax-ml/coix Inference Combinators in JAX | Jupyter Notebook | 53 | 3 | 1 years ago |
| 11 | jax-ml/jax-tpu-embedding | Python | 34 | 5 | Yesterday |
| 12 | jax-ml/australis | C++ | 28 | 5 | 3 years ago |
| 13 | jax-ml/jax-blog | 6 | 2 | 5 months ago |
Next step after the organization read
Open a flagship repository, compare a couple of portfolio leaders, or return to the organization map when you want a broader concentration read.
Learn and methodology
Keep trust-building context reachable, but behind the first data read instead of ahead of it.
How to read this organization snapshot
Total stars are useful as a discovery signal, but they do not tell you whether a team maintains every repository equally. Pair this page with release cadence, maintainer activity, and the flagship concentration shown above before making adoption decisions.
For broader background on GitStar's ranking logic and editorial guidance, see Methodology & Editorial Standards.