yuanchenyang / smalldiffusionLinks
Simple and readable code for training and sampling from diffusion models
☆643Updated 4 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.☆837Updated 4 months ago
- A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.☆948Updated last year
- Implementation of Diffusion Transformer (DiT) in JAX☆294Updated last year
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆612Updated last year
- From-scratch diffusion model implemented in PyTorch.☆98Updated last year
- Official repository for our work on micro-budget training of large-scale diffusion models.☆1,521Updated 9 months ago
- Text to Image Latent Diffusion using a Transformer core☆213Updated last year
- My take on Flow Matching☆80Updated 9 months ago
- UNet diffusion model in pure CUDA☆651Updated last year
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆426Updated 10 months ago
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆3,618Updated last month
- For optimization algorithm research and development.☆543Updated last week
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆1,054Updated 7 months ago
- Creating a diffusion model from scratch in PyTorch to learn exactly how they work.☆384Updated 6 months ago
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆320Updated 3 months ago
- A simple implimentation of Bayesian Flow Networks (BFN)☆240Updated last year
- Minimal Implementation of a D3PM in pytorch☆267Updated last year
- Official implementation of Inductive Moment Matching☆561Updated 3 months ago
- ☆516Updated 2 months ago
- TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…☆121Updated 8 months ago
- Dion optimizer algorithm☆374Updated last month
- Train VAE like a boss☆297Updated last year
- Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI☆291Updated 4 months ago
- Best practices & guides on how to write distributed pytorch training code☆526Updated last week
- ☆665Updated 10 months ago
- This repo contains the code for the paper "Intuitive physics understanding emerges fromself-supervised pretraining on natural videos"☆191Updated 8 months ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆778Updated 10 months ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆388Updated 2 months ago
- ☆619Updated 6 months ago
- Schedule-Free Optimization in PyTorch☆2,228Updated 5 months ago