A pipeline parallel training script for diffusion models that supports multiple state-of-the-art...