[JAX/Imagen] Imagen model with reference to Rosetta Github

sharathts · nv-kkudrynski · commit 6f3a71ae4676 · 2023-10-06T12:54:45.000-07:00
diff --git a/JAX/MultiModal/Imagen/README.md b/JAX/MultiModal/Imagen/README.md
@@ -0,0 +1,2 @@
+# Imagen on GPUs
+Please refer to [Rosetta Imagen](https://github.com/NVIDIA/JAX-Toolbox/tree/main/rosetta/rosetta/projects/imagen), NVIDIA's project that enables seamless training of LLMs, CV models and multimodal models in JAX, for information about running Imagen models and experiments on GPUs.

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,2 @@`
	`1`	`+# Imagen on GPUs`
	`2`	`+Please refer to [Rosetta Imagen](https://github.com/NVIDIA/JAX-Toolbox/tree/main/rosetta/rosetta/projects/imagen), NVIDIA's project that enables seamless training of LLMs, CV models and multimodal models in JAX, for information about running Imagen models and experiments on GPUs.`