PyPI page
Home page
Author:
JAX team
License:
Apache-2.0
Summary:
JAX Plugin for NVIDIA GPUs
Latest version:
0.12.0
Required dependencies:
jax-cuda13-pjrt
Optional dependencies:
nvidia-cublas
|
nvidia-cuda-cupti
|
nvidia-cuda-nvcc
|
nvidia-cuda-nvrtc
|
nvidia-cuda-runtime
|
nvidia-cudnn-cu13
|
nvidia-cufft
|
nvidia-cusolver
|
nvidia-cusparse
|
nvidia-nccl-cu13
|
nvidia-nvjitlink
|
nvidia-nvshmem-cu13
|
nvidia-nvvm
Downloads last day:
12,441
Downloads last week:
100,648
Downloads last month:
281,048
7-day smoothing