@arankomatsuzakiI'm trying to build this in JAX, which is essentially translating improved DDPM into JAX. Also trying to train it on a larger dataset.
I've alredy built VQ/VD/DC-VAE in JAX in the link below, but I've decided to swtich to diffusion models: t.co/E5sRbJC3O5
1 RT, 8 Fav2021/04/26 00:56
Societal Impacts of Artificial Intelligence with Miles Brundage - YouTube