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…☆31Updated 2 months ago
- ☆31Updated 2 years ago
- Code for Gaussian Score Matching Variational Inference☆35Updated 11 months ago
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆13Updated last year
- Public code for running Stochastic Gradient Descent on GPs.☆39Updated 9 months ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 4 years ago
- Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)☆42Updated 3 years ago
- Code for "Log Neural Controlled Differential Equations" (ICML 2024) and "Structured Linear CDEs" (NeurIPS 2025, Spotlight)☆29Updated 2 weeks ago
- Implementation of Action Matching for the Schrödinger equation☆25Updated 2 years ago
- Implementation of the work Variational multiple shooting for Bayesian ODEs with Gaussian processes☆13Updated 3 years ago
- ☆11Updated 2 years ago
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆42Updated 3 years ago
- Efficient Differentiable n-d PDE solvers in JAX.☆52Updated this week
- Example code to speed up GP inference with gradients for high-dimensional inputs.☆14Updated 4 years ago
- Simulation-based inference in JAX☆45Updated 2 months ago
- Normalizing Flows with a resampled base distribution☆47Updated 3 years ago
- Lightweight MCMC sampling for PyTorch Models aka My Corona Project☆52Updated 6 months ago
- Matrix-free linear algebra in JAX.☆159Updated 2 months ago
- Gradient Based Nested Sampling☆20Updated 2 years ago
- ☆52Updated 2 years ago
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆46Updated last year
- Gradient-informed particle MCMC methods☆12Updated 2 years ago
- Combination of transformers and diffusion models for flexible all-in-one simulation-based inference☆79Updated 7 months ago
- Kolmogorov–Arnold Networks built on JAX☆96Updated last week
- [NeurIPS 2020] Task-Agnostic Amortized Inference of Gaussian Process Hyperparameters (AHGP)☆23Updated 5 years ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆75Updated 4 months ago
- Conditional Flow Matching using JAX☆13Updated 2 years ago
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆57Updated 2 years ago
- Bayesian Neural Network Surrogates for Bayesian Optimization☆67Updated last year
- AL4PDE: A Benchmark for Active Learning for Neural PDE Solvers☆29Updated 8 months ago