yuanchenyang / smalldiffusion
Simple and readable code for training and sampling from diffusion models
☆490Updated 4 months ago
Alternatives and similar repositories for smalldiffusion
Users that are interested in smalldiffusion are comparing it to the libraries listed below
Sorting:
- Visualizations of the theory behind diffusion models.☆275Updated last week
- Implementation of Diffusion Transformer (DiT) in JAX☆275Updated 11 months ago
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆528Updated 10 months ago
- A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.☆734Updated last year
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆852Updated last month
- Official implementation of Inductive Moment Matching☆465Updated 2 months ago
- For optimization algorithm research and development.☆513Updated this week
- ☆492Updated 5 months ago
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆233Updated 2 months ago
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆2,537Updated 4 months ago
- Train VAE like a boss☆278Updated 6 months ago
- Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Models☆595Updated last month
- A simple implimentation of Bayesian Flow Networks (BFN)☆240Updated last year
- When it comes to optimizers, it's always better to be safe than sorry☆223Updated last month
- EDM2 and Autoguidance -- Official PyTorch implementation☆706Updated 5 months ago
- This repo contains the code for the paper "Intuitive physics understanding emerges fromself-supervised pretraining on natural videos"☆154Updated 2 months ago
- From-scratch diffusion model implemented in PyTorch.☆87Updated last year
- Muon optimizer: +>30% sample efficiency with <3% wallclock overhead☆623Updated last month
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆424Updated 5 months ago
- UNet diffusion model in pure CUDA☆602Updated 10 months ago
- Implementation of Autoregressive Diffusion in Pytorch☆377Updated 6 months ago
- Official repository for our work on micro-budget training of large-scale diffusion models.☆1,404Updated 4 months ago
- Annotated version of the Mamba paper☆483Updated last year
- Code and weights for the paper "Cluster and Predict Latents Patches for Improved Masked Image Modeling"☆102Updated last month
- DDPO for finetuning diffusion models, implemented in PyTorch with LoRA support☆569Updated last year
- A mini-library for training consistency models.☆243Updated last year
- A curated list for awesome discrete diffusion models resources.☆313Updated 3 months ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆286Updated 3 weeks ago
- Schedule-Free Optimization in PyTorch☆2,154Updated last month
- ☆538Updated last month