albcab / mfmLinks
Code to reproduce experiments in Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows
☆13Updated last year
Alternatives and similar repositories for mfm
Users that are interested in mfm are comparing it to the libraries listed below
Sorting:
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆13Updated last year
- Implementation of papers in 101 lines of code.☆18Updated 2 years ago
- Conditional Flow Matching using JAX☆13Updated 2 years ago
- Implementation of Action Matching for the Schrödinger equation☆25Updated 2 years ago
- Flow-matching algorithms in JAX☆114Updated last year
- ☆31Updated 2 years ago
- Stainless neural networks in JAX☆34Updated last month
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆71Updated 10 months ago
- Diffusion models in PyTorch☆124Updated this week
- Collecting research materials on neural samplers with diffusion/flow models☆59Updated 7 months ago
- Code for Gaussian Score Matching Variational Inference☆35Updated 11 months ago
- ☆24Updated last year
- Unofficial JAX implementation of the SOAP optimizer (https://arxiv.org/abs/2409.11321)☆24Updated 3 weeks ago
- Free-form flows are a generative model training a pair of neural networks via maximum likelihood☆50Updated 7 months ago
- ☆18Updated 7 months ago
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆26Updated 2 years ago
- Official code for "Improved Techniques for Maximum Likelihood Estimation for Diffusion ODEs" (ICML 2023)☆19Updated last week
- Einstein Fields official implementation in JAX.☆69Updated 5 months ago
- Likelihood-free AMortized Posterior Estimation with PyTorch☆132Updated last year
- Neural Diffusion Processes☆82Updated last year
- Modern normalizing flows in Python. Simple to use and easily extensible.☆11Updated 3 months ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆66Updated last year
- Implementation of Action Matching☆51Updated 2 years ago
- [ICML2022] Variational Wasserstein gradient flow☆24Updated 3 years ago
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆44Updated last year
- ☆52Updated 2 years ago
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆53Updated 6 months ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆48Updated last year
- Official implementation of "Lost in Latent Space: An Empirical Study of Latent Diffusion Models for Physics Emulation"☆52Updated 3 months ago
- ☆37Updated last week