Get Started

Technology · Library

JAX

Library · Google · Catalog tech@1

Composable transformations of numerical Python programs.

Categories
Matched as
  • 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

Momentum

Collecting0 of 5 signals

No score yet. Momentum is published only when at least 2 of the 5 signals have history. JAX has 0, so there is no single number: each signal is shown on its own below. It is not a zero, and nothing is estimated in its place.

Snapshot OCT 24, 2026 · Scoring tech@1

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

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

  • keras-team/keras

    Deep Learning for humans

    Tagged jaxPython

    64.3K

  • 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

  • kyegomez/OpenMythos

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

    Tagged jaxPython

    14.9K

  • 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

  • arogozhnikov/einops

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

    Tagged jaxPython

    9.6K

  • 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

  • HIPS/autograd

    Efficiently computes derivatives of NumPy code.

    Tagged jaxPython

    7.5K

  • sanchit-gandhi/whisper-jax

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

    Tagged jaxJupyter Notebook

    4.7K

  • tensorflow/datasets

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

    Tagged jaxPython

    4.6K

  • lululxvi/deepxde

    A library for scientific machine learning and physics-informed learning

    Tagged jaxPython

    4.4K

  • iree-org/iree

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

    Tagged jaxC++

    4K

  • danijar/dreamerv3

    Mastering Diverse Domains through World Models

    Tagged jaxPython

    3.8K

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

  • secretflow-lite

    SecretFlow Lite

    Depends on jaxPyPI

    Not published by PyPI
  • seqio

    SeqIO: Task-based datasets, preprocessing, and evaluation for sequence models.

    Depends on jaxPyPI

    Not published by PyPI
  • synthid-text

    SynthID Text: Identifying AI-generated text content

    Depends on jaxPyPI

    Not published by PyPI
  • tensorflow-federated

    An open-source framework for machine learning and other computations on decentralized data.

    Depends on jaxPyPI

    Not published by PyPI
  • tensorflow-quantum

    Library for hybrid quantum-classical machine learning.

    Depends on jaxPyPI

    Not published by PyPI
  • tensorflowjs

    Depends on jaxPyPI

    Not published by PyPI
  • tokamax

    A Pallas Custom Kernel Library.

    Depends on jaxPyPI

    Not published by PyPI
  • waymo-open-dataset-tf-2-12-0

    Depends on jaxPyPI

    Not published by PyPI

Latest news the latest event that mention it