yuanchenyang / smalldiffusionLinks
Simple and readable code for training and sampling from diffusion models
☆509Updated last week
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.☆771Updated last week
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆589Updated 11 months ago
- Implementation of Diffusion Transformer (DiT) in JAX☆278Updated last year
- A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.☆909Updated last year
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆2,867Updated 3 weeks ago
- Official implementation of Inductive Moment Matching☆488Updated 3 months ago
- ☆549Updated 6 months ago
- DDPO for finetuning diffusion models, implemented in PyTorch with LoRA support☆606Updated last year
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆898Updated 2 months ago
- From-scratch diffusion model implemented in PyTorch.☆93Updated last year
- Minimal Implementation of a D3PM in pytorch☆232Updated last year
- EDM2 and Autoguidance -- Official PyTorch implementation☆729Updated 6 months ago
- Train VAE like a boss☆281Updated 8 months ago
- For optimization algorithm research and development.☆521Updated this week
- Creating a diffusion model from scratch in PyTorch to learn exactly how they work.☆368Updated 2 months ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆309Updated last week
- A minimal yet resourceful implementation of diffusion models (along with pretrained models + synthetic images for nine datasets)☆286Updated 9 months ago
- A simple implimentation of Bayesian Flow Networks (BFN)☆240Updated last year
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆426Updated 6 months ago
- Text to Image Latent Diffusion using a Transformer core☆190Updated 9 months ago
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆381Updated 2 months ago
- TorchCFM: a Conditional Flow Matching library☆1,778Updated 3 months ago
- A summary of related works about flow matching, stochastic interpolants☆489Updated 3 months ago
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆251Updated 3 months ago
- ☆567Updated 2 months ago
- Annotated Flow Matching paper☆188Updated 9 months ago
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆370Updated 5 months ago
- [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think☆1,126Updated 3 months ago
- Annotated version of the Mamba paper☆485Updated last year
- [ICML 2024 Best Paper] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)☆586Updated last year