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 signalsNo 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
36.4K stars
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
1.3K stars
- pdebench/PDEBench
PDEBench: An Extensive Benchmark for Scientific Machine Learning
1.2K stars
- extropic-ai/thrml
Thermodynamic Hypergraphical Model Library in JAX
1.2K stars
- luchris429/purejaxrl
Really Fast End-to-End Jax RL Implementations
1.1K stars
- 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
1.1K stars
- google/jaxopt
Hardware accelerated, batchable and differentiable optimizers in JAX.
1.1K stars
- probml/dynamax
A Python package for probabilistic state space modeling with JAX
1K 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.
Not published by PyPI - jax-cuda12-pjrt
JAX XLA PJRT Plugin for NVIDIA GPUs
Not published by PyPI - jax-cuda12-plugin
JAX Plugin for NVIDIA GPUs
Not published by PyPI - jax-cuda13-pjrt
JAX XLA PJRT Plugin for NVIDIA GPUs
Not published by PyPI - jax-cuda13-plugin
JAX Plugin for NVIDIA GPUs
Not published by PyPI - jaxlib
XLA library for JAX
Not published by PyPI
Ecosystem 56packages that depend on it, most-adopted first
- optimistix
Nonlinear optimisation in JAX and Equinox.
Not published by PyPI - orbax-checkpoint
Orbax Checkpoint
Not published by PyPI - orbax-export
Orbax Export
Not published by PyPI - pathwaysutils
Package of Pathways-on-Cloud utilities.
Not published by PyPI - qiskit-addon-sqd
Classically postprocess noisy quantum samples to yield more accurate energy estimations
Not published by PyPI - qwix
Qwix is a Jax quantization library.
Not published by PyPI - rax
Learning-to-Rank using JAX.
Not published by PyPI - rlax
A library of reinforcement learning building blocks in JAX.
Not published by PyPI - rlinf-openpi
Physical Intelligence open source repo
Not published by PyPI - rpent-openpi
Physical Intelligence open source repo
Not published by PyPI - sax
Autograd and XLA for S-parameters
Not published by PyPI - secretflow
SecretFlow
Not published by PyPI
Latest news the latest event that mention it
- 21d agoOCT 3, 2026