PyTorch
image-generation
diffusion