yuanchenyang / smalldiffusionLinks
Simple and readable code for training and sampling from diffusion models
☆590Updated 2 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.☆811Updated 2 months ago
- A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.☆925Updated last year
- Implementation of Diffusion Transformer (DiT) in JAX☆291Updated last year
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆606Updated last year
- From-scratch diffusion model implemented in PyTorch.☆96Updated last year
- UNet diffusion model in pure CUDA☆648Updated last year
- For optimization algorithm research and development.☆530Updated this week
- ☆471Updated 3 weeks ago
- Minimal Implementation of a D3PM in pytorch☆249Updated last year
- Official repository for our work on micro-budget training of large-scale diffusion models.☆1,510Updated 7 months ago
- My take on Flow Matching☆71Updated 7 months ago
- Text to Image Latent Diffusion using a Transformer core☆201Updated last year
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆3,257Updated 3 months ago
- Official implementation of Inductive Moment Matching☆537Updated last month
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆428Updated 8 months ago
- Creating a diffusion model from scratch in PyTorch to learn exactly how they work.☆378Updated 4 months ago
- Rebuild the Stable Diffusion Model in a single python script. Tutorial for Harvard ML from Scratch Series☆214Updated 7 months ago
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆986Updated 4 months ago
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆284Updated last month
- Muon is an optimizer for hidden layers in neural networks☆1,595Updated last month
- Labs for MIT 6.S184/6.S975, IAP 2025☆193Updated 3 months ago
- Annotated version of the Mamba paper☆488Updated last year
- PyTorch Code for Energy-Based Transformers paper -- generalizable reasoning and scalable learning☆428Updated last month
- ☆599Updated 4 months ago
- Train VAE like a boss☆292Updated 10 months ago
- [ICML 2024 Best Paper] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)☆622Updated last year
- This repo contains the code for the paper "Intuitive physics understanding emerges fromself-supervised pretraining on natural videos"☆174Updated 6 months ago
- ☆266Updated 8 months ago
- Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI☆289Updated 2 months ago
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆399Updated 4 months ago