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: 179
- First release: 0.0 on 2018-12-07
- Latest release: 0.6.2 on 2025-06-17
Releases
PyPI Downloads
Dependencies
Jax has 11 dependencies, 6 of which optional.| Dependency | Optional |
|---|---|
| jaxlib | false |
| ml_dtypes | false |
| numpy | false |
| opt_einsum | false |
| scipy | false |
| jax-cuda12-plugin | true |
| jax-rocm60-plugin | true |
| kubernetes | true |
| libtpu | true |
| requests | true |
| xprof | true |