DoronHav / WassersteinFlowMatching
☆26Updated last month
Alternatives and similar repositories for WassersteinFlowMatching:
Users that are interested in WassersteinFlowMatching are comparing it to the libraries listed below
- Implementation of Action Matching☆41Updated last year
- ☆17Updated 3 years ago
- Code release for "Stochastic Optimal Control Matching"☆34Updated 8 months ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆57Updated last year
- Official implementation of Metric Flow Matching (NeurIPS 2024)☆35Updated 5 months ago
- Official implementation of Deep Momentum Schrödinger Bridge☆25Updated last year
- code for "Riemannian Flow Matching on General Geometries".☆224Updated last year
- A set of tests for evaluating large-scale algorithms for Wasserstein-2 transport maps computation (NeurIPS 2021)☆40Updated 2 years ago
- code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).☆68Updated last year
- ☆32Updated 11 months ago
- ☆19Updated 3 months ago
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆61Updated last month
- Implementation of Action Matching for the Schrödinger equation☆24Updated last year
- Score-based generative models for compact manifolds☆110Updated last year
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆54Updated 2 months ago
- ☆142Updated last year
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆24Updated last year
- Flow-matching algorithms in JAX☆90Updated 8 months ago
- Collecting research materials on neural samplers with diffusion/flow models☆46Updated last month
- ☆27Updated last year
- ☆30Updated last month
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems☆48Updated this week
- Neural Optimal Transport with Lagrangian Costs☆55Updated 9 months ago
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆32Updated 4 months ago
- Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)☆32Updated 3 years ago
- Code for the paper Iterated Denoising Energy Matching for Sampling from Boltzmann Densities.☆53Updated 3 weeks ago
- Euclidean Wasserstein-2 optimal transportation☆47Updated last year
- GP Sinkhorn Implementation, paper: https://www.mdpi.com/1099-4300/23/9/1134☆22Updated 3 years ago
- Simple Guidance Mechanisms for Discrete Diffusion Models☆36Updated 4 months ago
- The Superposition of Diffusion Models Using the Itô Density Estimator☆43Updated last month