necludov / action-matching
Implementation of Action Matching for the Schrödinger equation
☆24Updated last year
Alternatives and similar repositories for action-matching:
Users that are interested in action-matching are comparing it to the libraries listed below
- Implementation of Action Matching☆41Updated last year
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago
- ☆32Updated 11 months ago
- ☆52Updated 2 years ago
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆11Updated 10 months ago
- ☆19Updated 3 months ago
- NeurIPS'23: Energy Discrepancies: A Score-Independent Loss for Energy-Based Models☆16Updated 6 months ago
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆31Updated 4 months ago
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆37Updated 2 years ago
- ☆30Updated last month
- Normalizing Flows with a resampled base distribution☆46Updated 2 years ago
- Stochastic Normalizing Flows☆76Updated 3 years ago
- Code release for "Stochastic Optimal Control Matching"☆33Updated 8 months ago
- Library for normalizing flows and neural flows.☆24Updated 2 years ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆57Updated last year
- Collecting research materials on neural samplers with diffusion/flow models☆46Updated last month
- Implementation of papers in 101 lines of code.☆18Updated last year
- [ICLR 2022] Path integral sampler☆46Updated last year
- Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)☆32Updated 2 years ago
- Official implementation of Deep Momentum Schrödinger Bridge☆25Updated last year
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆24Updated last year
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆61Updated last month
- [NeurIPS 2020] Task-Agnostic Amortized Inference of Gaussian Process Hyperparameters (AHGP)☆21Updated 4 years ago
- Euclidean Wasserstein-2 optimal transportation☆47Updated last year
- ☆25Updated last month
- Perceiver (transformer variant) implemented in JAX and Flax☆12Updated 4 years ago
- Official repository of Implicit Neural Convolutional Kernels for Steerable CNNs, Zhdanov et al.☆28Updated 2 months ago
- orbital MCMC☆10Updated 3 years ago
- GP Sinkhorn Implementation, paper: https://www.mdpi.com/1099-4300/23/9/1134☆22Updated 3 years ago
- Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral☆47Updated 2 years ago