From 41ad89aa68be5194b4f5b4bfb8b1ace8c91d0806 Mon Sep 17 00:00:00 2001
From: Jeremy Berchtold <jberchtold@nvidia.com>
Date: Wed, 7 May 2025 10:26:19 -0700
Subject: [PATCH] Update README to list JAX NGC container

Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>
---
 README.rst | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/README.rst b/README.rst
index 8ce8dd7167..0a22ec1954 100644
--- a/README.rst
+++ b/README.rst
@@ -170,12 +170,20 @@ Docker (Recommended)
 ^^^^^^^^^^^^^^^^^^^
 The quickest way to get started with Transformer Engine is by using Docker images on
 `NVIDIA GPU Cloud (NGC) Catalog <https://catalog.ngc.nvidia.com/orgs/nvidia/containers/pytorch>`_.
+
+
 For example to use the NGC PyTorch container interactively,
 
 .. code-block:: bash
 
     docker run --gpus all -it --rm nvcr.io/nvidia/pytorch:25.04-py3
 
+For example to use the NGC JAX container interactively,
+
+.. code-block:: bash
+
+    docker run --gpus all -it --rm nvcr.io/nvidia/jax:25.04-py3
+
 Where 25.04 (corresponding to April 2025 release) is the container version.
 
 **Benefits of using NGC containers:**