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.4.28
Required dependencies: importlib-metadata | ml-dtypes | numpy | opt-einsum | scipy
Optional dependencies: jax-cuda12-plugin | jaxlib | libtpu-nightly | nvidia-cublas-cu12 | nvidia-cuda-cupti-cu12 | nvidia-cuda-nvcc-cu12 | nvidia-cuda-runtime-cu12 | nvidia-cudnn-cu12 | nvidia-cufft-cu12 | nvidia-cusolver-cu12 | nvidia-cusparse-cu12 | nvidia-nccl-cu12 | nvidia-nvjitlink-cu12 | protobuf | requests

Downloads last day: 229,330
Downloads last week: 1,496,496
Downloads last month: 6,767,995