JAX

library 0.9.2 ·python

✓ verified Jun 9, 2026 △ install stale

Composable transformations of Python+NumPy: differentiate (jax.grad), compile (jax.jit), vectorize (jax.vmap), parallelize (jax.shard_map). Current version is 0.9.2 (Mar 2026). Requires Python >=3.11. Install requires extras — bare pip install jax gives CPU-only minimal build.

Traffic · last 30 days · indexed Fri Mar 27 · updated Fri Jun 26

total hits 25

actors 5 distinct systems

last hit 2d ago AhrefsBot

ByteDance

MetaBot

Script

Humans

top countries 🇩🇪 Germany · 🇺🇸 United States · 🇨🇦 Canada · 🇸🇬 Singapore · 🇹🇷 Turkey

Resources

githubgithub.com/jax-ml/jax ↗

packagepypi.org/project/jax/ ↗

homepagejax.readthedocs.io ↗

API endpoints

full doc /v1/registry/jax

install /v1/registry/jax/install

compatibility /v1/registry/jax/compatibility