yuanchenyang / smalldiffusionLinks
Simple and readable code for training and sampling from diffusion models
☆573Updated 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.☆806Updated last month
- A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.☆924Updated last year
- Implementation of Diffusion Transformer (DiT) in JAX☆286Updated last year
- From-scratch diffusion model implemented in PyTorch.☆96Updated last year
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆602Updated last year
- UNet diffusion model in pure CUDA☆613Updated last year
- Official repository for our work on micro-budget training of large-scale diffusion models.☆1,503Updated 6 months ago
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆429Updated 7 months ago
- Text to Image Latent Diffusion using a Transformer core☆196Updated 11 months ago
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆3,153Updated 2 months ago
- For optimization algorithm research and development.☆525Updated last week
- My take on Flow Matching☆69Updated 6 months ago
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆274Updated 3 weeks ago
- Official implementation of Inductive Moment Matching☆529Updated 3 weeks ago
- Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI☆289Updated 2 months ago
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆948Updated 4 months ago
- Minimal Implementation of a D3PM in pytorch☆241Updated last year
- Train VAE like a boss☆287Updated 9 months ago
- ☆587Updated 3 months ago
- Muon is an optimizer for hidden layers in neural networks☆1,390Updated 3 weeks ago
- Creating a diffusion model from scratch in PyTorch to learn exactly how they work.☆373Updated 4 months ago
- PyTorch Code for Energy-Based Transformers paper -- generalizable reasoning and scalable learning☆400Updated 2 weeks ago
- A simple implimentation of Bayesian Flow Networks (BFN)☆239Updated last year
- ☆592Updated 8 months ago
- ☆447Updated 2 weeks ago
- Annotated version of the Mamba paper☆487Updated last year
- The boundary of neural network trainability is fractal☆215Updated last year
- Getting crystal-like representations with harmonic loss☆192Updated 4 months ago
- Best practices & guides on how to write distributed pytorch training code☆463Updated 5 months ago
- ☆261Updated 7 months ago