GitHub repository: Bhumika-Shetty/diffusion-pipe
Description: A pipeline parallel training script for diffusion models.