JAX

Momentum

Each signal is ranked as a percentile among the technologies that have it. A signal without enough history is not counted, and never shown as zero. How momentum works

Official repository stars

Collecting

Star growth needs repository snapshots 30 days apart.36,371 stars now

Member repository stars

Collecting

Star growth needs repository snapshots 30 days apart.0 new members in 30 days

Package adoption

—Not published by its registries

Its packages’ registries publish no adoption figure, so there is nothing to grow.

Open models

—No official models

No official model on Hugging Face, so this signal does not apply.

News

Collecting

News counts once there are 3 events across the last 60 days; it has 1.1 in the last 30 days · 0 in the 30 before

Adoption by registry

The growth of JAX’s own packages, one row per registry in that registry’s own unit. Registries count different things over different windows, so they are never added together.

No registry has two observations of JAX’s own packages about 30 days apart yet, so there is no growth to show. Nothing is estimated in its place.

News activity

Events · 7 days

0published events that mention it

Events · 30 days

1published events that mention it

Previous 30 days

0events in the 30 days before

Momentum history

Momentum history starts OCT 8, 2026. A chart appears once two daily snapshots carry a score; none does so far.

Repositories

Official 1

• jax-ml/jax

  Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

  Official repositoryPython

  36.4K stars

Members 43repositories tagged with one of its aliases, most-starred first

• keras-team/keras

  Deep Learning for humans

  Tagged jaxPython

  64.3K stars

• d2l-ai/d2l-en

  Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.

  Tagged jaxPython

  29.7K stars

• kyegomez/OpenMythos

  A theoretical reconstruction of the Claude Mythos architecture, built from first principles using the available research literature.

  Tagged jaxPython

  14.9K stars

• wandb/wandb

  The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.

  Tagged jaxPython

  11.3K stars

• arogozhnikov/einops

  Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

  Tagged jaxPython

  9.6K stars

• HenryNdubuaku/maths-cs-ai-compendium

  Become a cracked AI/ML researcher/engineer with this unconventional textbook covering maths, computing, and ML with intuition.

  Tagged jaxTypeScript

  7.6K stars

• HIPS/autograd

  Efficiently computes derivatives of NumPy code.

  Tagged jaxPython

  7.5K stars

• sanchit-gandhi/whisper-jax

  JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.

  Tagged jaxJupyter Notebook

  4.7K stars

• tensorflow/datasets

  TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...

  Tagged jaxPython

  4.6K stars

• lululxvi/deepxde

  A library for scientific machine learning and physics-informed learning

  Tagged jaxPython

  4.4K stars

• iree-org/iree

  A retargetable MLIR-based machine learning compiler and runtime toolkit.

  Tagged jaxC++

  4K stars

• danijar/dreamerv3

  Mastering Diverse Domains through World Models

  Tagged jaxPython

  3.8K stars

Packages

“Is” packages are JAX itself; ecosystem packages depend on them. The two are counted separately and never added together. Adoption is each registry’s own figure, with its percentile within that registry.

Is 6the technology itself, most-adopted first

• jax

  Differentiate, compile, and transform Numpy code.

  Listed in the catalogPyPI

  —Not published by PyPI
• jax-cuda12-pjrt

  JAX XLA PJRT Plugin for NVIDIA GPUs

  Published from jax-ml/jaxPyPI

  —Not published by PyPI
• jax-cuda12-plugin

  JAX Plugin for NVIDIA GPUs

  Published from jax-ml/jaxPyPI

  —Not published by PyPI
• jax-cuda13-pjrt

  JAX XLA PJRT Plugin for NVIDIA GPUs

  Published from jax-ml/jaxPyPI

  —Not published by PyPI
• jax-cuda13-plugin

  JAX Plugin for NVIDIA GPUs

  Published from jax-ml/jaxPyPI

  —Not published by PyPI
• jaxlib

  XLA library for JAX

  Published from jax-ml/jaxPyPI

  —Not published by PyPI

Ecosystem 56packages that depend on it, most-adopted first

• optimistix

  Nonlinear optimisation in JAX and Equinox.

  Depends on jaxPyPI

  —Not published by PyPI
• orbax-checkpoint

  Orbax Checkpoint

  Depends on jaxPyPI

  —Not published by PyPI
• orbax-export

  Orbax Export

  Depends on jaxPyPI

  —Not published by PyPI
• pathwaysutils

  Package of Pathways-on-Cloud utilities.

  Depends on jaxPyPI

  —Not published by PyPI
• qiskit-addon-sqd

  Classically postprocess noisy quantum samples to yield more accurate energy estimations

  Depends on jaxPyPI

  —Not published by PyPI
• qwix

  Qwix is a Jax quantization library.

  Depends on jaxPyPI

  —Not published by PyPI
• rax

  Learning-to-Rank using JAX.

  Depends on jaxPyPI

  —Not published by PyPI
• rlax

  A library of reinforcement learning building blocks in JAX.

  Depends on jaxPyPI

  —Not published by PyPI
• rlinf-openpi

  Physical Intelligence open source repo

  Depends on jaxPyPI

  —Not published by PyPI
• rpent-openpi

  Physical Intelligence open source repo

  Depends on jaxPyPI

  —Not published by PyPI
• sax

  Autograd and XLA for S-parameters

  Depends on jaxPyPI

  —Not published by PyPI
• secretflow

  SecretFlow

  Depends on jaxPyPI

  —Not published by PyPI

Latest news the latest event that mention it

• Reproducing Olmo 3 7B Pre-training in MaxText: case study of large scale training on TPUs1 source
  21d agoOCT 3, 2026