PyPI page
Home page
Author:
JAX team
License:
Apache-2.0
Summary:
Differentiate, compile, and transform Numpy code.
Latest version:
0.4.26
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:
217,988
Downloads last week:
1,365,356
Downloads last month:
5,688,448