PyPI page
Home page
Author:
JAX team
License:
Apache-2.0
Summary:
XLA library for JAX
Latest version:
0.4.25
Required dependencies:
ml-dtypes
|
numpy
|
scipy
Optional dependencies:
nvidia-cublas-cu11
|
nvidia-cublas-cu12
|
nvidia-cuda-cupti-cu11
|
nvidia-cuda-cupti-cu12
|
nvidia-cuda-nvcc-cu11
|
nvidia-cuda-nvcc-cu12
|
nvidia-cuda-runtime-cu11
|
nvidia-cuda-runtime-cu12
|
nvidia-cudnn-cu11
|
nvidia-cudnn-cu12
|
nvidia-cufft-cu11
|
nvidia-cufft-cu12
|
nvidia-cusolver-cu11
|
nvidia-cusolver-cu12
|
nvidia-cusparse-cu11
|
nvidia-cusparse-cu12
Downloads last day:
133,285
Downloads last week:
756,068
Downloads last month:
3,054,071