JAX docker update

prbasyal-amd · prbasyal-amd · commit 2295a192510d · 2025-12-03T12:01:14.000-05:00
diff --git a/docs/install/3rd-party/jax-install.rst b/docs/install/3rd-party/jax-install.rst
@@ -74,7 +74,7 @@ Docker image support
 
 AMD validates and publishes ready-made JAX images with ROCm backends on Docker
 Hub. The following Docker image tags and associated inventories are validated
-for ROCm 7.0.0.
+for ROCm 7.1.1.
 For ``jax-community`` images, see `rocm/jax-community <https://hub.docker.com/r/rocm/jax-community/tags>`__ on Docker Hub.
 
 .. tab-set::
@@ -90,25 +90,25 @@ For ``jax-community`` images, see `rocm/jax-community <https://hub.docker.com/r/
 
             .. code-block:: shell
 
-               docker pull rocm/jax:rocm7.1-jax0.7.1-py3.12
+               docker pull rocm/jax:rocm7.1.1-jax0.7.1-py3.12
 
             See
-            ``rocm/jax:rocm7.1-jax0.7.1-py3.12``
+            ``rocm/jax:rocm7.1.1-jax0.7.1-py3.12``
             on `Docker Hub
-            <https://hub.docker.com/layers/rocm/jax/rocm7.1-jax0.7.1-py3.12/images/sha256-bc661fee582c591c4a9355e249afb912e8fd7dfe39f1fc02008821b1dc739bdf>`__.
+            <https://hub.docker.com/layers/rocm/jax/rocm7.1.1-jax0.7.1-py3.12/images/sha256-1471f6853cde749d3bf06d35d94d7ca63d1fac5dbba9e6c884bb3cf6bb6f9ffa>`__.
 
          .. tab-item:: Python 3.11
 
             .. rubric:: Docker pull tag
 
             .. code-block:: shell
 
-               docker pull rocm/jax:rocm7.1-jax0.7.1-py3.11
+               docker pull rocm/jax:rocm7.1.1-jax0.7.1-py3.11
 
             See
-            ``rocm/jax:rocm7.1-jax0.7.1-py3.11``
+            ``rocm/jax:rocm7.1.1-jax0.7.1-py3.11``
             on `Docker Hub
-            <https://hub.docker.com/layers/rocm/jax/rocm7.1-jax0.7.1-py3.11/images/sha256-7e923ebfc117b2028c63991bfa1fbe274150ea94513c9530ece783f1a2f8a65e>`__.
+            <https://hub.docker.com/layers/rocm/jax/rocm7.1.1-jax0.7.1-py3.11/images/sha256-2359c120432df16e57468c52e6d164cc0db007cf385b97e48f2e244532176548>`__.
 
 .. _build-jax-rocm-docker-image:
 
