jax-cuda13-plugin0.7.2
jax-cuda13-plugin0.7.2
Published
JAX Plugin for NVIDIA GPUs
pip install jax-cuda13-plugin
Package Downloads
Authors
Project URLs
Requires Python
>=3.11
Dependencies
- jax-cuda13-pjrt
==0.7.2
- nvidia-cublas
>=13.0.0.19; sys_platform == "linux" and extra == "with-cuda"
- nvidia-cuda-cupti
>=13.0.48; sys_platform == "linux" and extra == "with-cuda"
- nvidia-cuda-nvcc
>=13.0.48; sys_platform == "linux" and extra == "with-cuda"
- nvidia-cuda-runtime
>=13.0.48; sys_platform == "linux" and extra == "with-cuda"
- nvidia-cudnn-cu13
<10.0,>=9.12.0.46; sys_platform == "linux" and extra == "with-cuda"
- nvidia-cufft
>=12.0.0.15; sys_platform == "linux" and extra == "with-cuda"
- nvidia-cusolver
>=12.0.3.29; sys_platform == "linux" and extra == "with-cuda"
- nvidia-cusparse
>=12.6.2.49; sys_platform == "linux" and extra == "with-cuda"
- nvidia-nccl-cu13
>=2.27.7; sys_platform == "linux" and extra == "with-cuda"
- nvidia-nvjitlink
>=13.0.39; sys_platform == "linux" and extra == "with-cuda"
- nvidia-cuda-nvrtc
>=13.0.48; sys_platform == "linux" and extra == "with-cuda"
- nvidia-nvshmem-cu13
>=3.3.20; sys_platform == "linux" and extra == "with-cuda"
- nvidia-nvvm
; extra == "with-cuda"