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.