Organization
jax-ml
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 (7), C++ (2), Jupyter Notebook (2)
Average size
3K
Stars per repository
Useful for distinguishing one flagship-heavy publisher from a repeatable portfolio.
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 92% of its visible star count.
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 (7), C++ (2), Jupyter Notebook (2). 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.5K | 3.5K | Today |
| 2 | jax-ml/scaling-book Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs | HTML | 935 | 134 | 1 months ago |
| 3 | jax-ml/jax-triton jax-triton contains integrations between JAX and OpenAI Triton | Python | 450 | 57 | 6 days ago |
| 4 | jax-ml/ml_dtypes A stand-alone implementation of several NumPy dtype extensions used in machine learning. | C++ | 342 | 56 | 2 weeks ago |
| 5 | jax-ml/oryx Oryx is a library for probabilistic programming and deep learning built on top of Jax. | Python | 314 | 13 | 1 weeks ago |
| 6 | jax-ml/jax-ai-stack | Python | 291 | 47 | 2 days ago |
| 7 | jax-ml/jax-llm-examples Minimal yet performant LLM examples in pure JAX | Python | 251 | 34 | 2 weeks ago |
| 8 | jax-ml/bayeux State of the art inference for your bayesian models. | Python | 240 | 12 | 3 months ago |
| 9 | jax-ml/bonsai Minimal, lightweight JAX implementations of popular models. | Jupyter Notebook | 234 | 43 | 1 months ago |
| 10 | jax-ml/coix Inference Combinators in JAX | Jupyter Notebook | 53 | 3 | 11 months ago |
| 11 | jax-ml/jax-tpu-embedding | Python | 31 | 5 | Today |
| 12 | jax-ml/australis | C++ | 28 | 4 | 3 years ago |
| 13 | jax-ml/jax-blog | 6 | 2 | 3 months ago |
Next step after the organization read
Learn and methodology
How to Read This 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.