PyPI page
Home page
Author:
JAX team
License:
Apache-2.0
Summary:
JAX Plugin for NVIDIA GPUs
Latest version:
0.9.0.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:
537
Downloads last week:
6,383
Downloads last month:
27,645