AaronHavens / FlowMatchingLinks
A repo where I play with conditional flow approaches for learning time-varying vector-fields.
☆19Updated last year
Alternatives and similar repositories for FlowMatching
Users that are interested in FlowMatching are comparing it to the libraries listed below
Sorting:
- The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)☆77Updated 6 months ago
- Code release for "Stochastic Optimal Control Matching"☆35Updated 10 months ago
- ☆32Updated last year
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆62Updated 3 months ago
- Light and Optimal Schrödinger Bridge Matching (ICML 2024) official PyTorch implementation☆41Updated 10 months ago
- Pytorch implementation of "Neural Optimal Transport with General Cost Functionals" (ICLR 2024)☆18Updated 9 months ago
- A demo shows how to combine Langevin dynamics with score matching for generative models.☆38Updated 4 years ago
- Educational implementation of the Discrete Flow Matching paper☆92Updated 9 months ago
- Official PyTorch implementation for the paper Minimizing Trajectory Curvature of ODE-based Generative Models, ICML 2023☆84Updated 4 months ago
- Minimum implementation of EDM (Elucidating the Design Space of Diffusion-Based Generative Models) on cifar10 and mnist☆54Updated last year
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆36Updated 3 weeks ago
- Official implementation of Metric Flow Matching (NeurIPS 2024)☆37Updated 7 months ago
- ☆31Updated 11 months ago
- A Tutorial for Diffusion Models☆52Updated last year
- ☆28Updated this week
- ☆44Updated last year
- Official code for "Improved Techniques for Maximum Likelihood Estimation for Diffusion ODEs" (ICML 2023)☆18Updated 3 months ago
- ☆40Updated last year
- Entropic Optimal Transport Benchmark (NeurIPS 2023).☆24Updated last year
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆55Updated 3 months ago
- Inference-Time Alignment in Protein Diffusion Models☆40Updated 5 months ago
- Official Repository for "Unbalancedness in Neural Monge Maps Improves Unpaired Domain Translation" [ICLR 2024]☆15Updated last year
- Official codebase for Exact Energy-Guided Diffusion Sampling via Contrastive Energy Prediction☆31Updated last year
- Official code for "Maximum Likelihood Training for Score-Based Diffusion ODEs by High-Order Denoising Score Matching" (ICML 2022)☆61Updated 2 years ago
- Pytorch implementation of "Light Schrödinger Bridge" (ICLR 2024)☆56Updated 10 months ago
- code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).☆71Updated last year
- Flow-matching algorithms in JAX☆97Updated 10 months ago
- Official Implementation of the paper: A Complete Recipe for Diffusion Generative Models☆30Updated 7 months ago
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆50Updated 11 months ago
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆24Updated last year