PyPI page
Home page
Author:
JAX team
License:
Apache-2.0
Summary:
JAX Plugin for NVIDIA GPUs
Latest version:
0.5.0
Required dependencies:
jax-cuda12-pjrt
Optional dependencies:
nvidia-cublas-cu12
|
nvidia-cuda-cupti-cu12
|
nvidia-cuda-nvcc-cu12
|
nvidia-cuda-runtime-cu12
|
nvidia-cudnn-cu12
|
nvidia-cufft-cu12
|
nvidia-cusolver-cu12
|
nvidia-cusparse-cu12
|
nvidia-nccl-cu12
|
nvidia-nvjitlink-cu12
Downloads last day:
15,644
Downloads last week:
82,825
Downloads last month:
216,011