Alternatives and similar repositories for adjoint_sampling

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

• smsharma / minified-generative-models
  Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…
  ☆133Updated last year
• maxxxzdn / erwin
  Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]
  ☆100Updated 2 months ago
• lollcat / fab-torch
  Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.
  ☆62Updated last year
• lucidrains / gotennet-pytorch
  Unofficial implementation of GotenNet, new SOTA 3d equivariant transformer, in Pytorch
  ☆64Updated 4 months ago
• juliusberner / sde_sampler
  Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)
  ☆65Updated 5 months ago
• facebookresearch / riemannian-fm
  code for "Riemannian Flow Matching on General Geometries".
  ☆251Updated last year
• google-deepmind / md4
  Official Jax Implementation of MD4 Masked Diffusion Models
  ☆123Updated 5 months ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆101Updated last year
• microsoft / soc-fine-tuning-sd
  Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching
  ☆40Updated 2 months ago
• jarridrb / DEM
  Code for the paper Iterated Denoising Energy Matching for Sampling from Boltzmann Densities.
  ☆59Updated 4 months ago
• HannesStark / dirichlet-flow-matching
  ☆118Updated last year
• necludov / super-diffusion
  The Superposition of Diffusion Models Using the Itô Density Estimator
  ☆52Updated 5 months ago
• nZhangx / TrajectoryFlowMatching
  Code repository for Trajectory Flow Matching
  ☆77Updated 9 months ago
• lazaratan / meta-flow-matching
  Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold
  ☆59Updated 5 months ago
• facebookresearch / SOC-matching
  Code release for "Stochastic Optimal Control Matching"
  ☆36Updated last year
• malbergo / stochastic-interpolants
  ☆171Updated last year
• liusulin / DDPD
  Official Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising" [ICLR 2025]
  ☆72Updated 4 months ago
• DoronHav / WassersteinFlowMatching
  ☆29Updated 2 weeks ago
• andrew-cr / discrete_flow_models
  Code for the paper https://arxiv.org/abs/2402.04997
  ☆91Updated last year
• martaskrt / fkc-diffusion
  ☆31Updated 2 months ago
• plainerman / Variational-Doob
  Lagrangian formulation of Doob's h-transform allowing for efficient rare event sampling
  ☆50Updated 5 months ago
• kuleshov-group / discrete-diffusion-guidance
  Simple Guidance Mechanisms for Discrete Diffusion Models
  ☆50Updated 8 months ago
• masa-ue / AlignInversePro
  Inference-Time Alignment in Protein Diffusion Models
  ☆42Updated 7 months ago
• nnaisense / bayesian-flow-networks
  This is the official code release for Bayesian Flow Networks.
  ☆291Updated last year
• maxxxzdn / clifford-group-equivariant-cnns
  Clifford-Steerable Convolutional Neural Networks [ICML'24]
  ☆49Updated 3 months ago
• lebellig / flow-matching
  Annotated Flow Matching paper
  ☆200Updated 11 months ago
• zacharyhorvitz / Fk-Diffusion-Steering
  A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.
  ☆176Updated 2 months ago
• lebellig / discrete-fm
  Educational implementation of the Discrete Flow Matching paper
  ☆106Updated last year
• necludov / jam
  Implementation of Action Matching
  ☆44Updated 2 years ago
• facebookresearch / generalized-schrodinger-bridge-matching
  code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).
  ☆75Updated last year