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.