jax-cuda12-pjrt

JAX XLA PJRT Plugin for NVIDIA GPUs

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install jax-cuda12-pjrt

Dependencies

  • None

Releases

Version Released Bullseye
Python 3.9
Bookworm
Python 3.11
Trixie
Python 3.13
Files
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-17
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-16
0.4.32 yanked 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-14

Issues with this package?

Page last updated 2025-11-18 19:54:19 UTC