yuanchenyang / smalldiffusionLinks
Simple and readable code for training and sampling from diffusion models
☆691Updated 7 months ago
Alternatives and similar repositories for smalldiffusion
Users that are interested in smalldiffusion are comparing it to the libraries listed below
Sorting:
- Interactive visualizations of the geometric intuition behind diffusion models.☆1,032Updated last week
- A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.☆979Updated last year
- Implementation of Diffusion Transformer (DiT) in JAX☆305Updated last year
- From-scratch diffusion model implemented in PyTorch.☆99Updated last year
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆630Updated last year
- Official repository for our work on micro-budget training of large-scale diffusion models.☆1,550Updated last year
- UNet diffusion model in pure CUDA☆662Updated last year
- Rebuild the Stable Diffusion Model in a single python script. Tutorial for Harvard ML from Scratch Series☆222Updated last year
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆347Updated 2 months ago
- A practical guide to diffusion models, implemented from scratch.☆242Updated last month
- ☆851Updated this week
- Text to Image Latent Diffusion using a Transformer core☆219Updated last year
- Minimal Implementation of a D3PM in pytorch☆285Updated last year
- For optimization algorithm research and development.☆556Updated 2 weeks ago
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆434Updated last year
- My take on Flow Matching☆90Updated last year
- Official implementation of Inductive Moment Matching☆570Updated 6 months ago
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆1,144Updated 2 months ago
- Creating a diffusion model from scratch in PyTorch to learn exactly how they work.☆392Updated 9 months ago
- ☆540Updated 5 months ago
- Best practices & guides on how to write distributed pytorch training code☆571Updated 3 months ago
- Universal Notation for Tensor Operations in Python.☆463Updated 9 months ago
- Labs for MIT 6.S184/6.S975, IAP 2025/2026☆301Updated last week
- Train VAE like a boss☆312Updated last year
- This repo implements Diffusion Transformers(DiT) in PyTorch and provides training and inference code on CelebHQ dataset☆58Updated last year
- A minimal yet resourceful implementation of diffusion models (along with pretrained models + synthetic images for nine datasets)☆305Updated last year
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆421Updated this week
- Muon is an optimizer for hidden layers in neural networks☆2,231Updated last week
- Annotated Flow Matching paper☆226Updated last year
- <Foundations of Computer Vision> Book☆423Updated 3 weeks ago