jax-cuda13-plugin 0.8.0

Author: JAX team

Requires Python: >=3.11

jax_cuda13_plugin-0.8.0-cp311-cp311-manylinux_2_27_aarch64.whl (5.2MiB)

jax_cuda13_plugin-0.8.0-cp311-cp311-manylinux_2_27_x86_64.whl (5.3MiB)

jax_cuda13_plugin-0.8.0-cp312-cp312-manylinux_2_27_aarch64.whl (5.2MiB)

jax_cuda13_plugin-0.8.0-cp312-cp312-manylinux_2_27_x86_64.whl (5.3MiB)

jax_cuda13_plugin-0.8.0-cp313-cp313-manylinux_2_27_aarch64.whl (5.2MiB)

jax_cuda13_plugin-0.8.0-cp313-cp313-manylinux_2_27_x86_64.whl (5.3MiB)

jax_cuda13_plugin-0.8.0-cp313-cp313t-manylinux_2_27_aarch64.whl (5.2MiB)

jax_cuda13_plugin-0.8.0-cp313-cp313t-manylinux_2_27_x86_64.whl (5.3MiB)

jax_cuda13_plugin-0.8.0-cp314-cp314-manylinux_2_27_aarch64.whl (5.2MiB)

jax_cuda13_plugin-0.8.0-cp314-cp314-manylinux_2_27_x86_64.whl (5.3MiB)

jax_cuda13_plugin-0.8.0-cp314-cp314t-manylinux_2_27_aarch64.whl (5.2MiB)

jax_cuda13_plugin-0.8.0-cp314-cp314t-manylinux_2_27_x86_64.whl (5.3MiB)

Dependencies:

jax-cuda13-pjrt (==0.8.0)

nvidia-cublas sys_platform == "linux" and extra == "with-cuda" (>=13.0.0.19)

nvidia-cuda-cupti sys_platform == "linux" and extra == "with-cuda" (>=13.0.48)

nvidia-cuda-nvcc sys_platform == "linux" and extra == "with-cuda" (>=13.0.48)

nvidia-cuda-runtime sys_platform == "linux" and extra == "with-cuda" (>=13.0.48)

nvidia-cudnn-cu13 sys_platform == "linux" and extra == "with-cuda" (<10.0,>=9.12.0.46)

nvidia-cufft sys_platform == "linux" and extra == "with-cuda" (>=12.0.0.15)

nvidia-cusolver sys_platform == "linux" and extra == "with-cuda" (>=12.0.3.29)

nvidia-cusparse sys_platform == "linux" and extra == "with-cuda" (>=12.6.2.49)

nvidia-nccl-cu13 sys_platform == "linux" and extra == "with-cuda" (>=2.27.7)

nvidia-nvjitlink sys_platform == "linux" and extra == "with-cuda" (>=13.0.39)

nvidia-cuda-nvrtc sys_platform == "linux" and extra == "with-cuda" (>=13.0.48)

nvidia-nvshmem-cu13 sys_platform == "linux" and extra == "with-cuda" (>=3.3.20)

nvidia-nvvm extra == "with-cuda"