yuanchenyang / smalldiffusionLinks
Simple and readable code for training and sampling from diffusion models
☆681Updated 6 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.☆923Updated this week
- A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.☆973Updated last year
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆625Updated last year
- Implementation of Diffusion Transformer (DiT) in JAX☆300Updated last year
- Official repository for our work on micro-budget training of large-scale diffusion models.☆1,546Updated 11 months ago
- From-scratch diffusion model implemented in PyTorch.☆99Updated last year
- UNet diffusion model in pure CUDA☆659Updated last year
- ☆810Updated 3 weeks ago
- Text to Image Latent Diffusion using a Transformer core☆217Updated last year
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆3,944Updated this week
- Official implementation of Inductive Moment Matching☆569Updated 5 months ago
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆340Updated last month
- My take on Flow Matching☆89Updated 11 months ago
- A practical guide to diffusion models, implemented from scratch.☆232Updated last week
- Best practices & guides on how to write distributed pytorch training code☆557Updated 2 months ago
- Minimal Implementation of a D3PM in pytorch☆281Updated last year
- Rebuild the Stable Diffusion Model in a single python script. Tutorial for Harvard ML from Scratch Series☆222Updated 11 months ago
- For optimization algorithm research and development.☆556Updated 2 weeks ago
- ☆537Updated 5 months ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆807Updated last year
- Train VAE like a boss☆311Updated last year
- Creating a diffusion model from scratch in PyTorch to learn exactly how they work.☆391Updated 9 months ago
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆1,129Updated last month
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆432Updated last year
- This repo contains the code for the paper "Intuitive physics understanding emerges fromself-supervised pretraining on natural videos"☆211Updated 10 months ago
- PyTorch Code for Energy-Based Transformers paper -- generalizable reasoning and scalable learning☆572Updated last month
- Schedule-Free Optimization in PyTorch☆2,244Updated 7 months ago
- Annotated version of the Mamba paper☆493Updated last year
- Dion optimizer algorithm☆413Updated this week
- LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.☆666Updated 4 months ago