jax-ml
jax-ml Joined Dec 2019 View on GitHub
- Stars
- 40,523
- across 15 public repos
- Rank
- #772
- of 94,827 organizations, top 0.8%
- Down 22 places this week
Repositories with 50+ stars
| Rank | Repository | Language | Stars |
|---|---|---|---|
| 911 | Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more | Python | 36,604 |
| 44,899 | Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs | SCSS | 1,525 |
| 131,417 | jax-triton contains integrations between JAX and OpenAI Triton | Python | 474 |
| 164,004 | A stand-alone implementation of several NumPy dtype extensions used in machine learning. | C++ | 369 |
| 182,294 | Oryx is a library for probabilistic programming and deep learning built on top of Jax. | Python | 327 |
| 188,389 | Python | 315 | |
| 203,996 | Minimal yet performant LLM examples in pure JAX | Python | 287 |
| 230,276 | State of the art inference for your bayesian models. | Python | 249 |
| 232,626 | Minimal, lightweight JAX implementations of popular models. | Python | 246 |
| 785,742 | Inference Combinators in JAX | Jupyter Notebook | 54 |
Among 94,827 organizations
40k–63k stars: 367 organizations, including jax-ml
Stars along the bottom, and organizations in each band on a log scale
Where the stars are
Languages
- Python 38.5k stars, 95% · 7 repos
- SCSS 1.5k stars, 4% · 1 repo
- C++ 369 stars, 1% · 1 repo
- Jupyter Notebook 54 stars, <1% · 1 repo
When they were made
10 repos with 50+ stars, made between 2018 and 2025
Quick stats
- Next rank
- +9 stars to reach #771
- Next milestone
- +150,246 stars to reach the top 0.1%
- Public repos
- 15, 10 with 50+ stars
- Popular repos
- 2 over 1k stars, 1 over 10k stars
- Top repo
- jax, #911 of all repos, 90% of stars
- Smaller repos
- 73 stars from repos under 50