diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index 44ac8e95d..4bbf966d5 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -292,6 +292,7 @@ jobs:
equinox
maxtext
levanter
+ equinox
upstream-t5x
upstream-pax
t5x
diff --git a/README.md b/README.md
index f5b419bc1..7d49431fc 100644
--- a/README.md
+++ b/README.md
@@ -13,7 +13,8 @@
- Components |
+ Components |
+ Container |
Build |
Test |
@@ -23,7 +24,9 @@
|
- |
+
+ ghcr.io/nvidia/jax:base
+ |
@@ -34,7 +37,9 @@
|
|
- |
+
+ ghcr.io/nvidia/jax:jax
+ |
@@ -55,7 +60,9 @@
|
|
- |
+
+ ghcr.io/nvidia/jax:levanter
+ |
@@ -69,7 +76,9 @@
|
|
- |
+
+ ghcr.io/nvidia/jax:equinox
+ |
@@ -83,7 +92,9 @@
|
|
- |
+
+ ghcr.io/nvidia/jax:triton
+ |
@@ -94,10 +105,12 @@
|
-
-
+ |
+
+ |
+
+ ghcr.io/nvidia/jax:upstream-t5x
|
- upstream |
@@ -107,7 +120,12 @@
|
- rosetta |
+
+
+ |
+
+ ghcr.io/nvidia/jax:t5x
+ |
@@ -117,10 +135,12 @@
|
-
-
+ |
+
+ |
+
+ ghcr.io/nvidia/jax:upstream-pax
|
- upstream |
@@ -130,7 +150,12 @@
|
- rosetta |
+
+
+ |
+
+ ghcr.io/nvidia/jax:pax
+ |
@@ -143,7 +168,9 @@
|
|
- |
+
+ ghcr.io/nvidia/jax:maxtext
+ |
@@ -167,6 +194,10 @@
+In all of the above cases, `ghcr.io/nvidia/jax:XXX` points to the most recent
+nightly build of the container for `XXX`. These containers are also tagged as
+`ghcr.io/nvidia/jax:XXX-YYYY-MM-DD`, if a stable reference is required.
+
## Note
This repo currently hosts a public CI for JAX on NVIDIA GPUs and covers some JAX libraries like: [T5x](https://github.com/google-research/t5x), [PAXML](https://github.com/google/paxml), [Transformer Engine](https://github.com/NVIDIA/TransformerEngine), [Pallas](https://jax.readthedocs.io/en/latest/pallas/quickstart.html) and others to come soon.
|