jax-cuda13-plugin@0.8.0

JAX Plugin for NVIDIA GPUs

2025-10-15 Python >=3.11 wheel
安装 (0.8.0)
pip install jax-cuda13-plugin==0.8.0
poetry add jax-cuda13-plugin==0.8.0
pipenv install jax-cuda13-plugin==0.8.0
conda install jax-cuda13-plugin=0.8.0
依赖 (14)
依赖
jax-cuda13-pjrt==0.10.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"
下载文件
文件名类型Python 版本大小
jax_cuda13_plugin-0.8.0-cp311-cp311-manylinux_2_27_aarch64.whl bdist_wheel cp311 5493483 bytes
jax_cuda13_plugin-0.8.0-cp311-cp311-manylinux_2_27_x86_64.whl bdist_wheel cp311 5509472 bytes
jax_cuda13_plugin-0.8.0-cp312-cp312-manylinux_2_27_aarch64.whl bdist_wheel cp312 5488819 bytes
jax_cuda13_plugin-0.8.0-cp312-cp312-manylinux_2_27_x86_64.whl bdist_wheel cp312 5507526 bytes
jax_cuda13_plugin-0.8.0-cp313-cp313-manylinux_2_27_aarch64.whl bdist_wheel cp313 5488998 bytes
jax_cuda13_plugin-0.8.0-cp313-cp313-manylinux_2_27_x86_64.whl bdist_wheel cp313 5507308 bytes
jax_cuda13_plugin-0.8.0-cp313-cp313t-manylinux_2_27_aarch64.whl bdist_wheel cp313 5500699 bytes
jax_cuda13_plugin-0.8.0-cp313-cp313t-manylinux_2_27_x86_64.whl bdist_wheel cp313 5516093 bytes
jax_cuda13_plugin-0.8.0-cp314-cp314-manylinux_2_27_aarch64.whl bdist_wheel cp314 5489256 bytes
jax_cuda13_plugin-0.8.0-cp314-cp314-manylinux_2_27_x86_64.whl bdist_wheel cp314 5508153 bytes
jax_cuda13_plugin-0.8.0-cp314-cp314t-manylinux_2_27_aarch64.whl bdist_wheel cp314 5501058 bytes
jax_cuda13_plugin-0.8.0-cp314-cp314t-manylinux_2_27_x86_64.whl bdist_wheel cp314 5516832 bytes
版本列表
0.10.2 2026-06-17
0.10.1 2026-05-20
0.10.0 2026-04-16
0.9.2 2026-03-18
0.9.1 2026-03-02
0.9.0.1 2026-02-05
0.9.0 2026-01-20
0.8.3 2026-01-29
0.8.2 2025-12-18
0.8.1 2025-11-18
0.8.0 2025-10-15
0.7.2 2025-09-16
0.0.1rc0 2025-07-15