Modular Diffusion

Python library built on PyTorch for designing and training your own diffusion models by mixing modules such as noise schedules, denoising networks and loss functions.