PyPI page
Home page
Author:
JAX team
License:
Apache-2.0
Summary:
JAX Plugin for NVIDIA GPUs
Latest version:
0.8.1
Required dependencies:
jax-cuda13-pjrt
|
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
Optional dependencies:
nvidia-nvvm
Downloads last day:
749
Downloads last week:
4,459
Downloads last month:
20,561