yuanchenyang / smalldiffusionLinks
Simple and readable code for training and sampling from diffusion models
☆511Updated last month
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.☆783Updated last month
- Implementation of Diffusion Transformer (DiT) in JAX☆279Updated last year
- A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.☆919Updated last year
- From-scratch diffusion model implemented in PyTorch.☆93Updated last year
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆599Updated last year
- UNet diffusion model in pure CUDA☆610Updated last year
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆426Updated 7 months ago
- Official repository for our work on micro-budget training of large-scale diffusion models.☆1,495Updated 6 months ago
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆923Updated 3 months ago
- For optimization algorithm research and development.☆521Updated this week
- Official implementation of Inductive Moment Matching☆515Updated this week
- Annotated version of the Mamba paper☆486Updated last year
- Minimal Implementation of a D3PM in pytorch☆238Updated last year
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆2,980Updated last month
- ☆573Updated 3 months ago
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆263Updated 4 months ago
- Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI☆287Updated last month
- When it comes to optimizers, it's always better to be safe than sorry☆302Updated 3 months ago
- Creating a diffusion model from scratch in PyTorch to learn exactly how they work.☆370Updated 3 months ago
- Muon is an optimizer for hidden layers in neural networks☆1,092Updated this week
- Schedule-Free Optimization in PyTorch☆2,189Updated last month
- Train VAE like a boss☆283Updated 8 months ago
- Text to Image Latent Diffusion using a Transformer core☆192Updated 10 months ago
- ☆273Updated last year
- My take on Flow Matching☆66Updated 6 months ago
- A simple implimentation of Bayesian Flow Networks (BFN)☆240Updated last year
- ☆323Updated this week
- Rebuild the Stable Diffusion Model in a single python script. Tutorial for Harvard ML from Scratch Series☆212Updated 5 months ago
- Single File, Single GPU, From Scratch, Efficient, Full Parameter Tuning library for "RL for LLMs"☆497Updated last week
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆385Updated 3 months ago