Package profile
jax
- Summary: Differentiate, compile, and transform Numpy code.
- Author: JAX team
- Homepage: https://github.com/jax-ml/jax
- Source: https://github.com/jax-ml/jax (Repo profile)
- Number of releases: 181
- First release: 0.0 on 2018-12-07
- Latest release: 0.7.1 on 2025-08-20
Releases
PyPI Downloads
Loading PyPI statistics...
Dependencies
Jax has 12 dependencies, 7 of which optional.| Dependency | Optional |
|---|---|
| jaxlib | false |
| ml_dtypes | false |
| numpy | false |
| opt_einsum | false |
| scipy | false |
| jax-cuda12-plugin | true |
| jax-cuda13-plugin | true |
| jax-rocm60-plugin | true |
| kubernetes | true |
| libtpu | true |
| requests | true |
| xprof | true |