tonyduan / diffusion
From-scratch diffusion model implemented in PyTorch.
☆87Updated last year
Alternatives and similar repositories for diffusion:
Users that are interested in diffusion are comparing it to the libraries listed below
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆285Updated 2 weeks ago
- Train VAE like a boss☆277Updated 6 months ago
- A mini-library for training consistency models.☆242Updated last year
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆120Updated 9 months ago
- Minimal Implementation of a D3PM in pytorch☆216Updated last year
- Focused on fast experimentation and simplicity☆72Updated 4 months ago
- The aim of this repository is to test and implement Flow-Matching-based models☆91Updated 3 months ago
- Implementation of Autoregressive Diffusion in Pytorch☆376Updated 6 months ago
- Annotated Flow Matching paper☆183Updated 7 months ago
- ☆178Updated 4 months ago
- Implementation of a multimodal diffusion transformer in Pytorch☆101Updated 10 months ago
- Consistency Models Made Easy☆277Updated 6 months ago
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆58Updated last week
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆157Updated last year
- My take on Flow Matching☆52Updated 3 months ago
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆519Updated 10 months ago
- Text to Image Latent Diffusion using a Transformer core☆181Updated 8 months ago
- Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public☆82Updated 2 months ago
- Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"☆210Updated 3 months ago
- A minimal implementation of a denoising diffusion model in PyTorch.☆122Updated 5 months ago
- Flow-matching algorithms in JAX☆90Updated 8 months ago
- FMBoost: Boosting Latent Diffusion with Flow Matching (ECCV 2024 Oral)☆226Updated 5 months ago
- ☆14Updated 3 months ago
- Implementation of the proposed Spline-Based Transformer from Disney Research☆88Updated 6 months ago
- The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)☆75Updated 4 months ago
- Implementation of Diffusion Transformer (DiT) in JAX☆272Updated 10 months ago
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆140Updated 2 months ago
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆351Updated 3 months ago
- Official PyTorch implementation of the paper: Flow Matching in Latent Space☆269Updated 3 months ago
- ☆283Updated 7 months ago