apple / ml-diffusion-alignment-gflownetLinks
PyTorch implementation for our paper "Improving GFlowNets for Text-to-Image Diffusion Alignment."
☆28Updated last year
Alternatives and similar repositories for ml-diffusion-alignment-gflownet
Users that are interested in ml-diffusion-alignment-gflownet are comparing it to the libraries listed below
Sorting:
- Official Implementation of understanding the latent space of diffusion models through the lens of riemannian geometry (NeurIPS 2023)☆91Updated last year
 - Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
 - Official Code Repository for the paper "Continuous Diffusion Model for Language Modeling" (NeurIPS 2025).☆46Updated last month
 - PyTorch code for FLD (Feature Likelihood Divergence), FID, KID, Precision, Recall, etc. using DINOv2, InceptionV3, CLIP, etc.☆59Updated last year
 - ☆38Updated last year
 - ☆48Updated last year
 - The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆124Updated last year
 - Code for ICLR 2023 Paper, "Stable Target Field for Reduced Variance Score Estimation in Diffusion Models”☆76Updated 2 years ago
 - Inference-Time Alignment in Protein Diffusion Models☆44Updated 9 months ago
 - Official PyTorch implementation for the paper Minimizing Trajectory Curvature of ODE-based Generative Models, ICML 2023☆88Updated 8 months ago
 - Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆46Updated 5 months ago
 - ☆44Updated last year
 - A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆194Updated 4 months ago
 - Official code for "Maximum Likelihood Training for Score-Based Diffusion ODEs by High-Order Denoising Score Matching" (ICML 2022)☆63Updated 3 years ago
 - ☆108Updated 2 years ago
 - Official PyTorch implementation for "Unifying Bayesian Flow Networks and Diffusion Models through Stochastic Differential Equations"☆41Updated last year
 - Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]☆90Updated 2 years ago
 - NeurIPS23 "Flow Factorized Representation Learning"☆39Updated last year
 - ICML 2023: Reduce, Reuse, Recycle: Composing Energy-Based Diffusion Models with MCMC☆146Updated last year
 - [ICML 2024]: Official implementation for the paper: "Consistent Diffusion Meets Tweedie"☆52Updated last year
 - Concept Learning Dynamics☆15Updated last year
 - Official implementation for Diffusion Alignment as Sampling (DAS), ICLR'25, Spotlight☆52Updated 8 months ago
 - Code for paper "Argmax Flows and Multinomial Diffusion: Learning Categorical Distributions"☆89Updated 4 years ago
 - Official Jax Implementation of MD4 Masked Diffusion Models☆138Updated 8 months ago
 - ☆144Updated last year
 - Official repo for Detecting, Explaining, and Mitigating Memorization in Diffusion Models (ICLR 2024)☆76Updated last year
 - Code for the paper: "Fine-Tuning Discrete Diffusion Models with Policy Gradient Methods"☆27Updated 5 months ago
 - ☆55Updated last year
 - Official implementation for Estimating the Optimal Covariance with Imperfect Mean in Diffusion Probabilistic Models (ICML 2022), and a re…☆109Updated 3 years ago
 - Official implementation of the paper The Hidden Language of Diffusion Models☆74Updated last year