yuanchenyang / smalldiffusionLinks
Simple and readable code for training and sampling from diffusion models
☆503Updated 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.☆730Updated last week
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆564Updated 11 months ago
- Implementation of Diffusion Transformer (DiT) in JAX☆276Updated 11 months ago
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆876Updated 2 months ago
- A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.☆741Updated last year
- From-scratch diffusion model implemented in PyTorch.☆92Updated last year
- Official repository for our work on micro-budget training of large-scale diffusion models.☆1,420Updated 4 months ago
- Official implementation of Inductive Moment Matching☆475Updated 2 months ago
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆2,699Updated last week
- ☆530Updated 6 months ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆719Updated 5 months ago
- Train VAE like a boss☆279Updated 7 months ago
- Annotated version of the Mamba paper☆482Updated last year
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆374Updated last month
- UNet diffusion model in pure CUDA☆606Updated 11 months ago
- [ICML 2024 Best Paper] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)☆581Updated last year
- A curated list for awesome discrete diffusion models resources.☆349Updated this week
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆423Updated 5 months ago
- A simple implimentation of Bayesian Flow Networks (BFN)☆240Updated last year
- Schedule-Free Optimization in PyTorch☆2,162Updated 2 weeks ago
- For optimization algorithm research and development.☆518Updated this week
- Annotated Flow Matching paper☆187Updated 8 months ago
- ☆267Updated 10 months ago
- A mini-library for training consistency models.☆244Updated last year
- TensorDict is a pytorch dedicated tensor container.☆925Updated last week
- A summary of related works about flow matching, stochastic interpolants☆466Updated 2 months ago
- ☆556Updated last month
- Minimal Implementation of a D3PM in pytorch☆226Updated last year
- Implementation of Autoregressive Diffusion in Pytorch☆382Updated 7 months ago
- LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.☆578Updated 7 months ago