PyPI Stats

Search

All packages
Top packages

Track packages

jax


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

Downloads last day: 202,280
Downloads last week: 2,195,801
Downloads last month: 8,864,296