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 Sat Jun 27

total hits 31

actors 8 distinct systems

last hit 1h ago ClaudeBot

ByteDance

5

GPTBot

4

MetaBot

4

Script

2

Amazonbot

1

ClaudeBot

1

Humans

5

top countries 🇺🇸 United States · 🇩🇪 Germany · 🇨🇦 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