tonyduan / diffusionLinks
From-scratch diffusion model implemented in PyTorch.
☆92Updated last year
Alternatives and similar repositories for diffusion
Users that are interested in diffusion are comparing it to the libraries listed below
Sorting:
- Train VAE like a boss☆279Updated 7 months ago
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆120Updated 10 months ago
- Implementation of Diffusion Transformer (DiT) in JAX☆276Updated 11 months ago
- A minimal implementation of a denoising diffusion model in PyTorch.☆122Updated 6 months ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆157Updated last year
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆294Updated last week
- Implementation of a multimodal diffusion transformer in Pytorch☆102Updated 11 months ago
- Annotated Flow Matching paper☆187Updated 8 months ago
- Flow-matching algorithms in JAX☆91Updated 9 months ago
- Focused on fast experimentation and simplicity☆73Updated 5 months ago
- A mini-library for training consistency models.☆244Updated last year
- Implementation of the proposed Spline-Based Transformer from Disney Research☆92Updated 6 months ago
- Consistency Models Made Easy☆284Updated 7 months ago
- Consistency models trained on CIFAR-10, in JAX.☆148Updated last year
- The aim of this repository is to test and implement Flow-Matching-based models☆92Updated 4 months ago
- ☆188Updated 5 months ago
- When it comes to optimizers, it's always better to be safe than sorry☆233Updated 2 months ago
- Diffusion models in PyTorch☆102Updated 2 months ago
- Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public☆83Updated 3 months ago
- My take on Flow Matching☆57Updated 4 months ago
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆361Updated 4 months ago
- [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule☆167Updated 2 months ago
- ☆88Updated last year
- Text to Image Latent Diffusion using a Transformer core☆184Updated 9 months ago
- Implementation of Autoregressive Diffusion in Pytorch☆382Updated 7 months ago
- Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"☆173Updated 11 months ago
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆63Updated last month
- Official implementation of Inductive Moment Matching☆475Updated 2 months ago
- ☆32Updated 11 months ago
- Minimal Implementation of a D3PM in pytorch☆226Updated last year