Sharath Turuvekere Sreenivas 6f3a71ae46 [JAX/Imagen] Imagen model with reference to Rosetta Github há 2 anos atrás
..
README.md 6f3a71ae46 [JAX/Imagen] Imagen model with reference to Rosetta Github há 2 anos atrás

README.md

Imagen on GPUs

Please refer to Rosetta 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.