CountDeckJoin the waitlist
Join the waitlist
Product
Live counters
Free tools
Compare
Best of
Rankings

jax-ml GitHub stars

A GitHub organization15 public repositories

jax-ml has 40,539 GitHub stars across 15 public repositories as of . Forks not counted. Source: GitHub's public API.

Total stars
40,539
Public repos
15
Most starred
36,617

jax

Most-starred jax-ml repositories

  1. jaxComposable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more36,617 ★
  2. scaling-bookHome for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs1,527 ★
  3. jax-tritonjax-triton contains integrations between JAX and OpenAI Triton475 ★
  4. ml_dtypesA stand-alone implementation of several NumPy dtype extensions used in machine learning.369 ★
  5. oryxOryx is a library for probabilistic programming and deep learning built on top of Jax.327 ★
  6. jax-ai-stackPython315 ★
  7. jax-llm-examplesMinimal yet performant LLM examples in pure JAX287 ★
  8. bayeuxState of the art inference for your bayesian models. 249 ★
  9. bonsaiMinimal, lightweight JAX implementations of popular models.246 ★
  10. coixInference Combinators in JAX54 ★
Show 5 more
  1. jax-tpu-embeddingHigh-performance large-scale embedding acceleration for JAX on Google TPU SparseCores.39 ★
  2. australisC++28 ★
  3. jax-blog6 ★
  4. .allstar0 ★
  5. .github0 ★

Questions

How many GitHub stars does jax-ml have?

jax-ml has 40,539 GitHub stars across 15 public repositories as of Nov 12, 2026, 20:14 UTC.

What is jax-ml's most-starred repository?

jax-ml/jax, with 36,617 stars, followed by jax-ml/scaling-book with 1,527.

How many public repositories does jax-ml have?

15 on GitHub, 15 of them its own (not forks).

Where does this number come from?

GitHub's public REST API: the stargazer count of every public repository the organization owns, summed. Forks are left out. CountDeck refreshes it every few hours and is not affiliated with GitHub.

Can I watch jax-ml's stars live?

Open any repository above for its live count, star history and next milestone. CountDeck also shows star counts as a Home Screen widget on iPhone and Android.