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-cuda12-pjrt
|
nvidia-cublas-cu12
|
nvidia-cuda-cupti-cu12
|
nvidia-cuda-nvcc-cu12
|
nvidia-cuda-nvrtc-cu12
|
nvidia-cuda-runtime-cu12
|
nvidia-cudnn-cu12
|
nvidia-cufft-cu12
|
nvidia-cusolver-cu12
|
nvidia-cusparse-cu12
|
nvidia-nccl-cu12
|
nvidia-nvjitlink-cu12
|
nvidia-nvshmem-cu12
Downloads last day:
27,562
Downloads last week:
117,862
Downloads last month:
522,850