jax-cuda13-pjrt

JAX XLA PJRT Plugin for NVIDIA GPUs

Installation

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

pip3 install jax-cuda13-pjrt

Dependencies

  • None

Releases

Version Released Bullseye
Python 3.9
Bookworm
Python 3.11
Trixie
Python 3.13
Files
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.0.1rc0 pre-release 2025-07-15

Issues with this package?

Page last updated 2026-03-18 23:33:12 UTC