Alternatives and similar repositories for diffusion

Users that are interested in diffusion are comparing it to the libraries listed below

• cloneofsimo / vqgan-training
  Train VAE like a boss
  ☆279Updated 7 months ago
• lucidrains / pytorch-custom-utils
  Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…
  ☆120Updated 10 months ago
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆276Updated 11 months ago
• filipbasara0 / simple-diffusion
  A minimal implementation of a denoising diffusion model in PyTorch.
  ☆122Updated 6 months ago
• louaaron / Reflected-Diffusion
  [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)
  ☆157Updated last year
• lucidrains / rectified-flow-pytorch
  Implementation of rectified flow and some of its followup research / improvements in Pytorch
  ☆294Updated last week
• lucidrains / multimodal-dit-pytorch
  Implementation of a multimodal diffusion transformer in Pytorch
  ☆102Updated 11 months ago
• lebellig / flow-matching
  Annotated Flow Matching paper
  ☆187Updated 8 months ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆91Updated 9 months ago
• fal-ai / diffusion-speedrun
  Focused on fast experimentation and simplicity
  ☆73Updated 5 months ago
• Kinyugo / consistency_models
  A mini-library for training consistency models.
  ☆244Updated last year
• lucidrains / spline-based-transformer
  Implementation of the proposed Spline-Based Transformer from Disney Research
  ☆92Updated 6 months ago
• locuslab / ect
  Consistency Models Made Easy
  ☆284Updated 7 months ago
• openai / consistency_models_cifar10
  Consistency models trained on CIFAR-10, in JAX.
  ☆148Updated last year
• leffff / FlowModels
  The aim of this repository is to test and implement Flow-Matching-based models
  ☆92Updated 4 months ago
• apple / ml-tarflow
  ☆188Updated 5 months ago
• kyleliang919 / C-Optim
  When it comes to optimizers, it's always better to be safe than sorry
  ☆233Updated 2 months ago
• probabilists / azula
  Diffusion models in PyTorch
  ☆102Updated 2 months ago
• lucidrains / hyper-connections
  Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public
  ☆83Updated 3 months ago
• dome272 / Flow-Matching
  My take on Flow Matching
  ☆57Updated 4 months ago
• lucidrains / mmdit
  Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch
  ☆361Updated 4 months ago
• NVlabs / GatedDeltaNet
  [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule
  ☆167Updated 2 months ago
• vdesai2014 / inference-optimization-blog-post
  ☆88Updated last year
• apapiu / transformer_latent_diffusion
  Text to Image Latent Diffusion using a Transformer core
  ☆184Updated 9 months ago
• lucidrains / autoregressive-diffusion-pytorch
  Implementation of Autoregressive Diffusion in Pytorch
  ☆382Updated 7 months ago
• lucidrains / titok-pytorch
  Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"
  ☆173Updated 11 months ago
• lucidrains / hl-gauss-pytorch
  The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch
  ☆63Updated last month
• lumalabs / imm
  Official implementation of Inductive Moment Matching
  ☆475Updated 2 months ago
• hyhieu / easy_pybind
  ☆32Updated 11 months ago
• cloneofsimo / d3pm
  Minimal Implementation of a D3PM in pytorch
  ☆226Updated last year