diff --git a/README.md b/README.md index 85adc7458..f98a0461c 100644 --- a/README.md +++ b/README.md @@ -247,6 +247,22 @@ + + + + + + + + ghcr.io/nvidia/jax:gemma + + + + + + + @@ -266,6 +282,7 @@ We currently enable training and evaluation for the following models: | [t5(t5x)](./rosetta/rosetta/projects/t5x) | ✔️ | ✔️ | ✔️ | | [ViT](./rosetta/rosetta/projects/vit) | ✔️ | ✔️ | ✔️ | | [Imagen](./rosetta/rosetta/projects/imagen) | ✔️ | | ✔️ | +| [PaliGemma](./rosetta/rosetta/projects/paligemma) | | ✔️ | ✔️ | We will update this table as new models become available, so stay tuned.