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
Oct 15, 2025
0.8.0
Sep 16, 2025
0.7.2
Aug 20, 2025
0.7.1
Jul 22, 2025
0.7.0
Jun 17, 2025
0.6.2
May 21, 2025
0.6.1
Apr 17, 2025
0.6.0
Mar 19, 2025
0.5.3
Feb 24, 2025
0.5.1
Jan 17, 2025
0.5.0
Dec 17, 2024
0.4.38
Dec 05, 2024
0.4.36
Oct 22, 2024
0.4.35
Oct 04, 2024
0.4.34
Sep 16, 2024
0.4.33
Sep 11, 2024
0.4.32
Jul 30, 2024
0.4.31
Jun 18, 2024
0.4.30
Jun 10, 2024
0.4.29
May 09, 2024
0.4.28
May 07, 2024
0.4.27
Apr 03, 2024
0.4.26
Feb 26, 2024
0.4.25
Feb 07, 2024
0.4.24
Dec 14, 2023
0.4.23
Dec 14, 2023
0.4.22
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 |