jax-cuda12-plugin@0.7.2

JAX Plugin for NVIDIA GPUs

2025-09-16 Python >=3.11 wheel

pip install jax-cuda12-plugin==0.7.2

poetry add jax-cuda12-plugin==0.7.2

pipenv install jax-cuda12-plugin==0.7.2

conda install jax-cuda12-plugin=0.7.2

依赖
`jax-cuda12-pjrt==0.10.2`
`nvidia-cublas-cu12>=12.1.3.1; sys_platform == "linux" and extra == "with-cuda"`
`nvidia-cuda-cupti-cu12>=12.1.105; sys_platform == "linux" and extra == "with-cuda"`
`nvidia-cuda-nvcc-cu12>=12.6.85; sys_platform == "linux" and extra == "with-cuda"`
`nvidia-cuda-runtime-cu12>=12.1.105; sys_platform == "linux" and extra == "with-cuda"`
`nvidia-cudnn-cu12<10.0,>=9.8; sys_platform == "linux" and extra == "with-cuda"`
`nvidia-cufft-cu12>=11.0.2.54; sys_platform == "linux" and extra == "with-cuda"`
`nvidia-cusolver-cu12>=11.4.5.107; sys_platform == "linux" and extra == "with-cuda"`
`nvidia-cusparse-cu12>=12.1.0.106; sys_platform == "linux" and extra == "with-cuda"`
`nvidia-nccl-cu12>=2.18.1; sys_platform == "linux" and extra == "with-cuda"`
`nvidia-nvjitlink-cu12>=12.1.105; sys_platform == "linux" and extra == "with-cuda"`
`nvidia-cuda-nvrtc-cu12>=12.1.55; sys_platform == "linux" and extra == "with-cuda"`
`nvidia-nvshmem-cu12>=3.2.5; sys_platform == "linux" and extra == "with-cuda"`

文件名	类型	Python 版本	大小
jax_cuda12_plugin-0.7.2-cp311-cp311-manylinux_2_27_aarch64.whl	bdist_wheel	`cp311`	5466351 bytes
jax_cuda12_plugin-0.7.2-cp311-cp311-manylinux_2_27_x86_64.whl	bdist_wheel	`cp311`	5479806 bytes
jax_cuda12_plugin-0.7.2-cp312-cp312-manylinux_2_27_aarch64.whl	bdist_wheel	`cp312`	5460119 bytes
jax_cuda12_plugin-0.7.2-cp312-cp312-manylinux_2_27_x86_64.whl	bdist_wheel	`cp312`	5476864 bytes
jax_cuda12_plugin-0.7.2-cp313-cp313-manylinux_2_27_aarch64.whl	bdist_wheel	`cp313`	5460772 bytes
jax_cuda12_plugin-0.7.2-cp313-cp313-manylinux_2_27_x86_64.whl	bdist_wheel	`cp313`	5476949 bytes
jax_cuda12_plugin-0.7.2-cp313-cp313t-manylinux_2_27_aarch64.whl	bdist_wheel	`cp313`	5474728 bytes
jax_cuda12_plugin-0.7.2-cp313-cp313t-manylinux_2_27_x86_64.whl	bdist_wheel	`cp313`	5485236 bytes
jax_cuda12_plugin-0.7.2-cp314-cp314-manylinux_2_27_aarch64.whl	bdist_wheel	`cp314`	5461974 bytes
jax_cuda12_plugin-0.7.2-cp314-cp314-manylinux_2_27_x86_64.whl	bdist_wheel	`cp314`	5477722 bytes
jax_cuda12_plugin-0.7.2-cp314-cp314t-manylinux_2_27_aarch64.whl	bdist_wheel	`cp314`	5475355 bytes
jax_cuda12_plugin-0.7.2-cp314-cp314t-manylinux_2_27_x86_64.whl	bdist_wheel	`cp314`	5485605 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.7.1 2025-08-20

0.7.0 2025-07-22

0.6.2 2025-06-17

0.6.1 2025-05-21

0.6.0 2025-04-17

0.5.3 2025-03-19

0.5.1 2025-02-24

0.5.0 2025-01-23

0.4.38 2024-12-17

0.4.36 2024-12-05

0.4.35 2024-10-22

0.4.34 2024-10-04

0.4.33 2024-09-25

0.4.32 2024-09-11

0.4.31 2024-07-30

0.4.30 2024-06-18

0.4.29 2024-06-10

0.4.28 2024-05-09

0.4.27 2024-05-07

0.4.26 2024-04-03

0.4.25 2024-02-26

0.4.24 2024-02-07

0.4.23 2023-12-15

0.4.22 2023-12-14