Alternatives and similar repositories for Diffusion

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

• CW-Huang / sdeflow-light
  A minimalist implementation of score-based diffusion model
  ☆127Updated 3 years ago
• AlexiaJM / score_sde_fast_sampling
  Repository for the "Gotta Go Fast When Generating Data with Score-Based Models" paper
  ☆104Updated 3 years ago
• juliusberner / sde_sampler
  Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)
  ☆62Updated 2 months ago
• yang-song / score_flow
  Official code for "Maximum Likelihood Training of Score-Based Diffusion Models", NeurIPS 2021 (spotlight)
  ☆146Updated 3 years ago
• google-research / vdm
  ☆315Updated 2 years ago
• malbergo / stochastic-interpolants
  ☆146Updated last year
• enhuiz / langevin-dynamics-and-score-matching
  A demo shows how to combine Langevin dynamics with score matching for generative models.
  ☆38Updated 4 years ago
• JTT94 / diffusion_schrodinger_bridge
  PyTorch Implementation of DSB for Score Based Generative Modeling. Experiments managed using Hydra.
  ☆168Updated 3 years ago
• crowsonkb / jax-wavelets
  The 2D discrete wavelet transform for JAX
  ☆43Updated 2 years ago
• Jackson-Kang / Pytorch-Diffusion-Model-Tutorial
  A simple tutorial of Diffusion Probabilistic Models
  ☆92Updated 6 months ago
• ghliu / SB-FBSDE
  Likelihood Training of Schrödinger Bridge using FBSDEs Theory, ICLR 2022
  ☆84Updated 3 years ago
• addtt / variational-diffusion-models
  PyTorch implementation of Variational Diffusion Models.
  ☆82Updated last year
• nv-tlabs / CLD-SGM
  Score-Based Generative Modeling with Critically-Damped Langevin Diffusion
  ☆197Updated last year
• noegroup / stochastic_normalizing_flows
  Stochastic Normalizing Flows
  ☆76Updated 3 years ago
• Kinyugo / consistency_models
  A mini-library for training consistency models.
  ☆244Updated last year
• swyoon / pytorch-energy-based-model
  Simple illustrative examples for energy-based models in PyTorch
  ☆62Updated 5 years ago
• louaaron / Reflected-Diffusion
  [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)
  ☆157Updated last year
• yiyixuxu / denoising-diffusion-flax
  Implementing the Denoising Diffusion Probabilistic Model in Flax
  ☆148Updated 2 years ago
• VSehwag / minimal-diffusion
  A minimal yet resourceful implementation of diffusion models (along with pretrained models + synthetic images for nine datasets)
  ☆286Updated 9 months ago
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆50Updated 10 months ago
• yuyang-shi / dsbm-pytorch
  PyTorch Implementation of Diffusion Schrodinger Bridge Matching
  ☆158Updated 3 months ago
• necludov / jam
  Implementation of Action Matching
  ☆44Updated 2 years ago
• MaximeRobeyns / bayesian_flow_networks
  A PyTorch implementation of Bayesian flow networks (Graves et al., 2023).
  ☆26Updated last year
• andrew-cr / tauLDR
  Code for the paper https://arxiv.org/abs/2205.14987v2
  ☆51Updated last year
• chrischute / flowplusplus
  Implementation of Flow++ in PyTorch
  ☆40Updated 5 years ago
• michaelsdr / sinkformers
  Transformers with doubly stochastic attention
  ☆45Updated 2 years ago
• ermongroup / sliced_score_matching
  Code for reproducing results in the sliced score matching paper (UAI 2019)
  ☆147Updated 5 years ago
• AaltoML / generative-inverse-heat-dissipation
  Code release for the paper Generative Modeling With Inverse Heat Dissipation
  ☆71Updated 2 years ago
• smsharma / minified-generative-models
  Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…
  ☆130Updated last year
• lucidrains / jax2torch
  Use Jax functions in Pytorch
  ☆242Updated last year