From b96e0d193bba86a13ae075e84e6e1df4dfefe43e Mon Sep 17 00:00:00 2001
From: Istvan Kiss <istvan.kiss@amd.com>
Date: Thu, 20 Nov 2025 17:42:21 +0100
Subject: [PATCH] Add JAX Plugin-PJRT support table 7.0.1 (#5673)

* Add JAX Plugin-PJRT support table

* Update wordlist.txt for JAX comp table
---
 .wordlist.txt                                   |  2 ++
 .../ml-compatibility/jax-compatibility.rst      | 17 +++++++++++++++++
 2 files changed, 19 insertions(+)

diff --git a/.wordlist.txt b/.wordlist.txt
index 13b18adb2..01d42ac6a 100644
--- a/.wordlist.txt
+++ b/.wordlist.txt
@@ -243,6 +243,7 @@ Intersphinx
 Intra
 Ioffe
 JAX's
+JAXLIB
 Jinja
 JSON
 Jupyter
@@ -375,6 +376,7 @@ perf
 PEQT
 PIL
 PILImage
+PJRT
 POR
 PRNG
 PRs
diff --git a/docs/compatibility/ml-compatibility/jax-compatibility.rst b/docs/compatibility/ml-compatibility/jax-compatibility.rst
index 479256c48..3ab64c46e 100644
--- a/docs/compatibility/ml-compatibility/jax-compatibility.rst
+++ b/docs/compatibility/ml-compatibility/jax-compatibility.rst
@@ -47,6 +47,23 @@ with ROCm support:
    `Community ROCm JAX Docker images <https://hub.docker.com/r/rocm/jax-community>`_
    follow upstream JAX releases and use the latest available ROCm version.
 
+JAX Plugin-PJRT with JAX/JAXLIB compatibility
+================================================================================
+
+Portable JIT Runtime (PJRT) is an open, stable interface for device runtime and
+compiler. The following table details the ROCm version compatibility matrix
+between JAX Plugin–PJRT and JAX/JAXLIB.
+
+.. list-table::
+    :header-rows: 1
+
+    * - JAX Plugin-PJRT
+      - JAX/JAXLIB
+      - ROCm
+    * - 0.6.0
+      - 0.6.2, 0.6.0
+      - 7.0.1, 7.0.0
+
 Use cases and recommendations
 ================================================================================