jax-cuda12-plugin@0.4.23

JAX Plugin for NVIDIA GPUs

2023-12-15 Python >=3.11 wheel

pip install jax-cuda12-plugin==0.4.23

poetry add jax-cuda12-plugin==0.4.23

pipenv install jax-cuda12-plugin==0.4.23

conda install jax-cuda12-plugin=0.4.23

依赖
`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.4.23-cp310-cp310-manylinux2014_aarch64.whl	bdist_wheel	`cp310`	12384142 bytes
jax_cuda12_plugin-0.4.23-cp310-cp310-manylinux2014_x86_64.whl	bdist_wheel	`cp310`	12311874 bytes
jax_cuda12_plugin-0.4.23-cp311-cp311-manylinux2014_aarch64.whl	bdist_wheel	`cp311`	12383994 bytes
jax_cuda12_plugin-0.4.23-cp311-cp311-manylinux2014_x86_64.whl	bdist_wheel	`cp311`	12312085 bytes
jax_cuda12_plugin-0.4.23-cp312-cp312-manylinux2014_aarch64.whl	bdist_wheel	`cp312`	12381422 bytes
jax_cuda12_plugin-0.4.23-cp312-cp312-manylinux2014_x86_64.whl	bdist_wheel	`cp312`	12310480 bytes
jax_cuda12_plugin-0.4.23-cp39-cp39-manylinux2014_aarch64.whl	bdist_wheel	`cp39`	12385291 bytes
jax_cuda12_plugin-0.4.23-cp39-cp39-manylinux2014_x86_64.whl	bdist_wheel	`cp39`	12312027 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