PyPI page
Home page
Author:
JAX team
License:
Apache-2.0
Summary:
JAX Plugin for NVIDIA GPUs
Latest version:
0.11.2
Required dependencies:
winjax-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:
2
Downloads last week:
23
Downloads last month:
181
7-day smoothing