Explorar el Código

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

Sharath Turuvekere Sreenivas hace 2 años
padre
commit
6f3a71ae46
Se han modificado 1 ficheros con 2 adiciones y 0 borrados
  1. 2 0
      JAX/MultiModal/Imagen/README.md

+ 2 - 0
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.