mnist-diffusion-jax
Reimplementing some diffusion + related papers as I read them!
Initial to-do list:
- U-Net: Convolutional Networks for Biomedical Image Segmentation. Olaf Ronneberger, Philipp Fischer, Thomas Brox. https://arxiv.org/abs/1505.04597
- Denoising Diffusion Probabilistic Models. Jonathan Ho, Ajay Jain, Pieter Abbeel. https://arxiv.org/abs/2006.11239
- Denoising Diffusion Implicit Models. Jiaming Song, Chenlin Meng, Stefano Ermon. https://arxiv.org/abs/2006.11239
- Improved Denoising Diffusion Probabilistic Models. Alex Nichol, Prafulla Dhariwal. https://arxiv.org/abs/2102.09672
- Classifier-Free Diffusion Guidance. Jonathan Ho & Tim Salimans. https://arxiv.org/abs/2207.12598