GitHub repository: tdrussell/diffusion-pipe
Primary language: Python
Description: A pipeline parallel training script for diffusion models.