Alternatives and similar repositories for diffusion-finetuning

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

• microsoft / soc-fine-tuning-sd
  Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching
  ☆55Updated 6 months ago
• GFNOrg / gfn-diffusion
  ☆35Updated 8 months ago
• zdhNarsil / Diffusion-Generative-Flow-Samplers
  PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…
  ☆26Updated last year
• d-tiapkin / gflownet-rl
  Repository for "Generative Flow Networks as Entropy-Regularized RL" (AISTATS-2024, Oral)
  ☆40Updated last year
• ngushchin / EntropicOTBenchmark
  Entropic Optimal Transport Benchmark (NeurIPS 2023).
  ☆27Updated last year
• google-deepmind / md4
  Official Jax Implementation of MD4 Masked Diffusion Models
  ☆146Updated 9 months ago
• facebookresearch / SOC-matching
  Code release for "Stochastic Optimal Control Matching"
  ☆39Updated last year
• ling-pan / GAFN
  ☆25Updated last year
• kuleshov-group / discrete-diffusion-guidance
  Simple Guidance Mechanisms for Discrete Diffusion Models
  ☆60Updated 11 months ago
• harryjo97 / RDLM
  Official Code Repository for the paper "Continuous Diffusion Model for Language Modeling" (NeurIPS 2025).
  ☆46Updated 2 months ago
• juliusberner / sde_sampler
  Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)
  ☆70Updated 8 months ago
• AI4Science-WestlakeU / flow_guidance
  [ICML 2025] The official implementation of the paper "On the Guidance of Flow Matching"
  ☆64Updated 2 months ago
• thu-ml / CEP-energy-guided-diffusion
  Official codebase for Exact Energy-Guided Diffusion Sampling via Contrastive Energy Prediction
  ☆35Updated 2 years ago
• vdutor / neural-diffusion-processes
  Neural Diffusion Processes
  ☆81Updated last year
• timgaripov / compositional-sculpting
  Code for paper "Compositional Sculpting of Iterative Generative Processes"
  ☆25Updated 2 years ago
• ozekri / SEPO
  Code for the paper: "Fine-Tuning Discrete Diffusion Models with Policy Gradient Methods"
  ☆28Updated 6 months ago
• GFNOrg / GFlowNet-EM
  Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.
  ☆41Updated last year
• zhaoyl18 / SEIKO
  SEIKO is a novel reinforcement learning method to efficiently fine-tune diffusion models in an online setting. Our methods outperform all…
  ☆27Updated last year
• liusulin / DDPD
  Official Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising" [ICLR 2025]
  ☆83Updated 7 months ago
• masa-ue / AlignInversePro
  Inference-Time Alignment in Protein Diffusion Models
  ☆47Updated 10 months ago
• AllanYangZhou / nfn
  NF-Layers for constructing neural functionals.
  ☆91Updated last year
• LuChengTHU / mle_score_ode
  Official code for "Maximum Likelihood Training for Score-Based Diffusion ODEs by High-Order Denoising Score Matching" (ICML 2022)
  ☆64Updated 3 years ago
• lebellig / discrete-fm
  Educational implementation of the Discrete Flow Matching paper
  ☆126Updated last year
• hankook / CLEL
  ☆16Updated 2 years ago
• dbsxodud-11 / ls_gfn
  Official Code for Local Search GFlowNets (ICLR 2024 Spotlight)
  ☆23Updated 9 months ago
• DenisBless / variational_sampling_methods
  [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".
  ☆40Updated 11 months ago
• tung-nd / TNP-pytorch
  Official implementation of Transformer Neural Processes
  ☆78Updated 3 years ago
• didriknielsen / argmax_flows
  Code for paper "Argmax Flows and Multinomial Diffusion: Learning Categorical Distributions"
  ☆89Updated 4 years ago
• kuleshov-group / remdm
  Remasking Discrete Diffusion Models with Inference-Time Scaling
  ☆57Updated 8 months ago
• facebookresearch / generalized-schrodinger-bridge-matching
  code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).
  ☆82Updated last year