JAX Plugin for NVIDIA GPUs
Project Links
Meta
Author: JAX team
Requires Python: >=3.11
Classifiers
Development Status
- 5 - Production/Stable
Programming Language
- Python :: 3.11
- Python :: 3.12
- Python :: 3.13
- Python :: 3.14
- Python :: Free Threading :: 3 - Stable
Description not available
0.8.0
Oct 15, 2025
0.7.2
Sep 16, 2025
0.7.1
Aug 20, 2025
0.7.0
Jul 22, 2025
0.6.2
Jun 17, 2025
0.6.1
May 21, 2025
0.6.0
Apr 17, 2025
0.5.3
Mar 19, 2025
0.5.1
Feb 24, 2025
0.5.0
Jan 17, 2025
0.4.38
Dec 17, 2024
0.4.36
Dec 05, 2024
0.4.35
Oct 22, 2024
0.4.34
Oct 04, 2024
0.4.33
Sep 16, 2024
0.4.32
Sep 11, 2024
0.4.31
Jul 30, 2024
0.4.30
Jun 18, 2024
0.4.29
Jun 10, 2024
0.4.28
May 09, 2024
0.4.27
May 07, 2024
0.4.26
Apr 03, 2024
0.4.25
Feb 26, 2024
0.4.24
Feb 07, 2024
0.4.23
Dec 14, 2023
0.4.22
Dec 14, 2023
Wheel compatibility matrix
| Platform | CPython 3.11 | CPython 3.12 | CPython 3.13 | CPython 3.14 | CPython (additional flags: t) 3.13 | CPython (additional flags: t) 3.14 |
|---|---|---|---|---|---|---|
| manylinux_2_27_aarch64 | ||||||
| manylinux_2_27_x86_64 |
Files in release
jax_cuda12_plugin-0.8.0-cp311-cp311-manylinux_2_27_aarch64.whl (5.3MiB)
jax_cuda12_plugin-0.8.0-cp311-cp311-manylinux_2_27_x86_64.whl (5.3MiB)
jax_cuda12_plugin-0.8.0-cp312-cp312-manylinux_2_27_aarch64.whl (5.2MiB)
jax_cuda12_plugin-0.8.0-cp312-cp312-manylinux_2_27_x86_64.whl (5.3MiB)
jax_cuda12_plugin-0.8.0-cp313-cp313-manylinux_2_27_aarch64.whl (5.2MiB)
jax_cuda12_plugin-0.8.0-cp313-cp313-manylinux_2_27_x86_64.whl (5.3MiB)
jax_cuda12_plugin-0.8.0-cp313-cp313t-manylinux_2_27_aarch64.whl (5.3MiB)
jax_cuda12_plugin-0.8.0-cp313-cp313t-manylinux_2_27_x86_64.whl (5.3MiB)
jax_cuda12_plugin-0.8.0-cp314-cp314-manylinux_2_27_aarch64.whl (5.2MiB)
jax_cuda12_plugin-0.8.0-cp314-cp314-manylinux_2_27_x86_64.whl (5.3MiB)
jax_cuda12_plugin-0.8.0-cp314-cp314t-manylinux_2_27_aarch64.whl (5.3MiB)
jax_cuda12_plugin-0.8.0-cp314-cp314t-manylinux_2_27_x86_64.whl (5.3MiB)
