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

@@ -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

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
 ================================================================================