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

  • DBraun/DawDreamer

    Digital Audio Workstation with Python; VST instruments/effects, parameter automation, FAUST, JAX, Warp Markers, and JUCE processors

    Tagged jaxC++

    1.3K

  • pdebench/PDEBench

    PDEBench: An Extensive Benchmark for Scientific Machine Learning

    Tagged jaxPython

    1.2K

  • extropic-ai/thrml

    Thermodynamic Hypergraphical Model Library in JAX

    Tagged jaxPython

    1.2K

  • luchris429/purejaxrl

    Really Fast End-to-End Jax RL Implementations

    Tagged jaxPython

    1.1K

  • Toni-SM/skrl

    Modular Reinforcement Learning (RL) library (implemented in PyTorch, JAX, and NVIDIA Warp) with support for Gymnasium/Gym, NVIDIA Isaac Lab, MuJoCo Playground and other environments

    Tagged jaxPython

    1.1K

  • google/jaxopt

    Hardware accelerated, batchable and differentiable optimizers in JAX.

    Tagged jaxPython

    1.1K

  • probml/dynamax

    A Python package for probabilistic state space modeling with JAX

    Tagged jaxPython

    1K

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

  • aqtp

    Accurate Quantized Training library.

    Depends on jaxPyPI

    Not published by PyPI
  • augmax

    Efficiently Composable Data Augmentation on the GPU with Jax

    Depends on jaxPyPI

    Not published by PyPI
  • blackjax

    Flexible and fast sampling in Python

    Depends on jaxPyPI

    Not published by PyPI
  • blackjax-nightly

    Flexible and fast sampling in Python

    Depends on jaxPyPI

    Not published by PyPI
  • chex

    Chex: Testing made fun, in JAX!

    Depends on jaxPyPI

    Not published by PyPI
  • clu

    Set of libraries for ML training loops in JAX.

    Depends on jaxPyPI

    Not published by PyPI
  • cuequivariance-jax

    CUDA accelerated equivariant operations

    Depends on jaxPyPI

    Not published by PyPI
  • diffrax

    GPU+autodiff-capable ODE/SDE/CDE solvers written in JAX.

    Depends on jaxPyPI

    Not published by PyPI
  • distrax

    Distrax: Probability distributions in JAX.

    Depends on jaxPyPI

    Not published by PyPI
  • drjax

    DrJAX - Scalable and Differentiable MapReduce Primitives in JAX.

    Depends on jaxPyPI

    Not published by PyPI
  • dynamax

    Dynamic State Space Models in JAX.

    Depends on jaxPyPI

    Not published by PyPI
  • e3nn-jax

    Equivariant convolutional neural networks for the group E(3) of 3 dimensional rotations, translations, and mirrors.

    Depends on jaxPyPI

    Not published by PyPI

Latest news the latest event that mention it