PyPI page
Home page
Author:
Oleg Eterevsky
Summary:
Native Windows CUDA support for JAX (unofficial PJRT plugin loader)
Latest version:
0.11.2
Required dependencies:
jax
|
jaxlib
|
nvidia-cublas
|
nvidia-cuda-cupti
|
nvidia-cuda-nvcc
|
nvidia-cuda-nvrtc
|
nvidia-cuda-runtime
|
nvidia-cudnn-cu13
|
nvidia-cufft
|
nvidia-cusolver
|
nvidia-cusparse
|
nvidia-nvjitlink
|
nvidia-nvvm
|
winjax-cuda13-pjrt
|
winjax-cuda13-plugin
Downloads last day:
4
Downloads last week:
20
Downloads last month:
98
7-day smoothing