jaxlib has 16.8M monthly downloads and 36K GitHub stars. Version 0.10.2 supports Python 3.11+, first published in 2018.
It is actively maintained under a permissive license, last released in Jun 2026.
It is built on top of ml-dtypes, numpy and scipy. It is used by 759 tracked packages, including jax, tensorflowjs and optax.
GitHub topics include jax.
Package Insights
((week_daily_avg - month_daily_avg) / month_daily_avg) * 100Weekly Downloads
GitHub Stars
Downloads by OS
Python Versions
Top Countries
Used By
- jax
- tensorflowjs
- optax
- chex
- numpyro
- clu
- orbax-export
- e3nn-jax
- jax-md
- jaxopt
- qwix
- aqtp
- tokamax
- seqio
- drjax
- seqio-nightly
- distrax
- blackjax
- rax
- klujax
- tpu-inference
- mujoco-mjx
- jax-dataclasses
- dopamine-rl
- brax
- rlax
- sella
- dinosaur
- qrisp
- qiskit-addon-sqd
- tensorflow-federated
- dynamax
- mctx
- scib-metrics
- pennylane-catalyst
- jax-mps
- gpjax
- inferactively-pymdp
- jaxns
- simsopt
- eigh
- slsqp-jax
- searchlibrium
- qpax
- enzyme-ad
- diff-biophys
- ananke-causal
- jaxsim
- pyhgf
- unxt