facebookresearch / SOC-matchingLinks
Code release for "Stochastic Optimal Control Matching"
☆35Updated 11 months ago
Alternatives and similar repositories for SOC-matching
Users that are interested in SOC-matching are comparing it to the libraries listed below
Sorting:
- [ICLR 2022] Path integral sampler☆47Updated last year
- ☆32Updated last year
- Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral☆48Updated 2 years ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆58Updated last year
- ☆35Updated 3 months ago
- [ICML2022] Variational Wasserstein gradient flow☆23Updated 2 years ago
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆62Updated 4 months ago
- Neural Diffusion Processes☆81Updated 11 months ago
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆35Updated 7 months ago
- ☆28Updated 3 weeks ago
- Implementation of Action Matching☆44Updated 2 years ago
- Likelihood Training of Schrödinger Bridge using FBSDEs Theory, ICLR 2022☆84Updated 3 years ago
- Implementation of Action Matching for the Schrödinger equation☆24Updated 2 years ago
- ☆28Updated last year
- ☆20Updated 5 months ago
- A minimalist implementation of score-based diffusion model☆127Updated 3 years ago
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆24Updated last year
- JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".☆30Updated 3 months ago
- Collecting research materials on neural samplers with diffusion/flow models☆54Updated last week
- ☆24Updated last year
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆38Updated last month
- Euclidean Wasserstein-2 optimal transportation☆47Updated last year
- Implementing Controlled Monte Carlo Diffusions (ICLR 2024)☆15Updated 9 months ago
- A set of tests for evaluating large-scale algorithms for Wasserstein-2 transport maps computation (NeurIPS 2021)☆42Updated 3 years ago
- Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)☆33Updated 3 years ago
- code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"☆107Updated last month
- ☆158Updated last year
- ☆24Updated last year
- NeurIPS'23: Energy Discrepancies: A Score-Independent Loss for Energy-Based Models☆16Updated 8 months ago
- Flow-matching algorithms in JAX☆97Updated 11 months ago