pub jax

Signed-off-by: tylertitsworth <tyler.titsworth@intel.com>

Author: tylertitsworth

.github/dockerhub-readmes.json

@@ -43,6 +43,10 @@
         {
             "fname": "./workflows/README.md",
             "repo-name": "intel/ai-workflows"
+        },
+        {
+            "fname": "./jax/README.md",
+            "repo-name": "intel/intel-optimized-xla"
         }
     ]
 }

jax/README.md

@@ -10,6 +10,18 @@ The images below include [JAX\*] and [Intel® Extension for OpenXLA\*].
 | -------------------------- | --------- | ----------------- | -------- | --------------- |
 | `0.4.0-pip-base`, `latest` | [v0.4.32] | [v0.4.0-jax]      | [v0.9.0] | [v0.4.0]        |
 
+### Run the Container
+
+```bash
+docker run -it --rm \
+    --device /dev/dri \
+    -v /dev/dri/by-path:/dev/dri/by-path \
+    --net=host \
+    -v $PWD/workspace:/workspace \
+    -w /workspace \
+    intel/intel-optimized-xla:latest
+```
+
 The images below additionally include [Jupyter Notebook](https://jupyter.org/) server:
 
 | Tag(s)              | [JAX\*]   | [Intel® Extension for OpenXLA\*] | [Flax]   | Dockerfile      |
@@ -20,6 +32,8 @@ The images below additionally include [Jupyter Notebook](https://jupyter.org/) s
 
 ```bash
 docker run -it --rm \
+    --device /dev/dri \
+    -v /dev/dri/by-path:/dev/dri/by-path \
     -p 8888:8888 \
     --net=host \
     -v $PWD/workspace:/workspace \