New Libraries > flaxdiff

Framework and Library for building and training Diffusion models in multi-node multi-device distributed settings (TPUs)