yuanchenyang / smalldiffusionLinks
Simple and readable code for training and sampling from diffusion models
☆624Updated 3 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.☆825Updated 3 months ago
- A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.☆937Updated last year
- Implementation of Diffusion Transformer (DiT) in JAX☆292Updated last year
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆611Updated last year
- Official repository for our work on micro-budget training of large-scale diffusion models.☆1,512Updated 8 months ago
- From-scratch diffusion model implemented in PyTorch.☆98Updated last year
- UNet diffusion model in pure CUDA☆649Updated last year
- ☆508Updated 2 months ago
- Creating a diffusion model from scratch in PyTorch to learn exactly how they work.☆381Updated 6 months ago
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆303Updated 2 months ago
- Minimal Implementation of a D3PM in pytorch☆261Updated last year
- Text to Image Latent Diffusion using a Transformer core☆209Updated last year
- Official implementation of Inductive Moment Matching☆550Updated 2 months ago
- Rebuild the Stable Diffusion Model in a single python script. Tutorial for Harvard ML from Scratch Series☆217Updated 8 months ago
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆3,493Updated 2 weeks ago
- This repo contains the code for the paper "Intuitive physics understanding emerges fromself-supervised pretraining on natural videos"☆186Updated 7 months ago
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆1,034Updated 6 months ago
- My take on Flow Matching☆78Updated 8 months ago
- For optimization algorithm research and development.☆539Updated 2 weeks ago
- Train VAE like a boss☆293Updated 11 months ago
- PyTorch Code for Energy-Based Transformers paper -- generalizable reasoning and scalable learning☆524Updated 2 weeks ago
- ☆609Updated 5 months ago
- Schedule-Free Optimization in PyTorch☆2,217Updated 4 months ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆771Updated 10 months ago
- ☆281Updated 9 months ago
- Dion optimizer algorithm☆361Updated last week
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆381Updated 2 months ago
- A minimal implementation of DeepMind's Genie world model☆951Updated last week
- Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI☆291Updated 4 months ago
- Annotated version of the Mamba paper☆489Updated last year