A pipeline parallel training script for diffusion models. - View it on GitHub