jax-cuda13-plugin@0.9.0.1

JAX Plugin for NVIDIA GPUs

2026-02-05 Python >=3.11 wheel

pip install jax-cuda13-plugin==0.9.0.1

poetry add jax-cuda13-plugin==0.9.0.1

pipenv install jax-cuda13-plugin==0.9.0.1

conda install jax-cuda13-plugin=0.9.0.1

依赖
`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.9.0.1-cp311-cp311-manylinux_2_27_aarch64.whl	bdist_wheel	`cp311`	5626557 bytes
jax_cuda13_plugin-0.9.0.1-cp311-cp311-manylinux_2_27_x86_64.whl	bdist_wheel	`cp311`	5626779 bytes
jax_cuda13_plugin-0.9.0.1-cp312-cp312-manylinux_2_27_aarch64.whl	bdist_wheel	`cp312`	5621127 bytes
jax_cuda13_plugin-0.9.0.1-cp312-cp312-manylinux_2_27_x86_64.whl	bdist_wheel	`cp312`	5624250 bytes
jax_cuda13_plugin-0.9.0.1-cp313-cp313-manylinux_2_27_aarch64.whl	bdist_wheel	`cp313`	5621760 bytes
jax_cuda13_plugin-0.9.0.1-cp313-cp313-manylinux_2_27_x86_64.whl	bdist_wheel	`cp313`	5623750 bytes
jax_cuda13_plugin-0.9.0.1-cp313-cp313t-manylinux_2_27_aarch64.whl	bdist_wheel	`cp313`	5634668 bytes
jax_cuda13_plugin-0.9.0.1-cp313-cp313t-manylinux_2_27_x86_64.whl	bdist_wheel	`cp313`	5632456 bytes
jax_cuda13_plugin-0.9.0.1-cp314-cp314-manylinux_2_27_aarch64.whl	bdist_wheel	`cp314`	5622126 bytes
jax_cuda13_plugin-0.9.0.1-cp314-cp314-manylinux_2_27_x86_64.whl	bdist_wheel	`cp314`	5624871 bytes
jax_cuda13_plugin-0.9.0.1-cp314-cp314t-manylinux_2_27_aarch64.whl	bdist_wheel	`cp314`	5634677 bytes
jax_cuda13_plugin-0.9.0.1-cp314-cp314t-manylinux_2_27_x86_64.whl	bdist_wheel	`cp314`	5633116 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