瀏覽代碼

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

Sharath Turuvekere Sreenivas 2 年之前
父節點
當前提交
6f3a71ae46
共有 1 個文件被更改,包括 2 次插入0 次删除
  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.