NickHunt-Smith / MCMC-diffusionLinks
A Metropolis-Hastings MCMC sampler accelerated via diffusion models
☆16Updated last year
Alternatives and similar repositories for MCMC-diffusion
Users that are interested in MCMC-diffusion are comparing it to the libraries listed below
Sorting:
- Autoencoders in Function Space: Resolution-invariant autoencoders for functional data based on the functional variational autoencoder (FV…☆29Updated last week
- ☆30Updated 2 years ago
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆12Updated last year
- Code for Gaussian Score Matching Variational Inference☆35Updated 9 months ago
- ☆10Updated 2 years ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 4 years ago
- Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)☆39Updated 3 years ago
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆41Updated 2 years ago
- Implementation of Action Matching for the Schrödinger equation☆25Updated 2 years ago
- Gradient-informed particle MCMC methods☆12Updated last year
- Code for "Log Neural Controlled Differential Equations" (ICML 2024) and "Structured Linear CDEs" (NeurIPS 2025, Spotlight)☆27Updated last month
- Conditional Flow Matching using JAX☆13Updated 2 years ago
- Matrix-free linear algebra in JAX.☆151Updated this week
- ☆52Updated 2 years ago
- Public code for running Stochastic Gradient Descent on GPs.☆39Updated 7 months ago
- Lightweight MCMC sampling for PyTorch Models aka My Corona Project☆51Updated 3 months ago
- GATSBI: Generative Adversarial Training for Simulation-Based Inference☆19Updated 2 years ago
- Gradient Based Nested Sampling☆20Updated last year
- The code enables to perform Bayesian inference in an efficient manner through the use of Hamiltonian Neural Networks (HNNs), Deep Neural …☆16Updated 2 years ago
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆55Updated 2 years ago
- Efficient Differentiable n-d PDE solvers in JAX.☆52Updated last month
- Implementation of the work Variational multiple shooting for Bayesian ODEs with Gaussian processes☆13Updated 3 years ago
- ☆22Updated last year
- Simulation-based inference in JAX☆40Updated 2 weeks ago
- Example code to speed up GP inference with gradients for high-dimensional inputs.☆14Updated 4 years ago
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆50Updated 4 months ago
- Normalizing Flows with a resampled base distribution☆47Updated 3 years ago
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆40Updated 11 months ago
- A Python library for variational inference with normalizing flow and annealing☆15Updated 8 months ago
- An unofficial implementation of the Fourier Neural Operator in Flax☆18Updated last year