jax

PyPI page
Home page
Author: JAX team
License: Apache-2.0
Summary: Differentiate, compile, and transform Numpy code.
Latest version: 0.5.3
Required dependencies: jaxlib | ml_dtypes | numpy | opt_einsum | scipy
Optional dependencies: jax-cuda12-plugin | jax-rocm60-plugin | jaxlib | kubernetes | libtpu | requests

Downloads last day: 188,745
Downloads last week: 1,828,146
Downloads last month: 8,829,091