NickHunt-Smith / MCMC-diffusion
A Metropolis-Hastings MCMC sampler accelerated via diffusion models
☆13Updated 8 months ago
Alternatives and similar repositories for MCMC-diffusion:
Users that are interested in MCMC-diffusion are comparing it to the libraries listed below
- ☆24Updated last year
- Efficient Differentiable n-d PDE solvers in JAX.☆27Updated 4 months ago
- Gradient-informed particle MCMC methods☆11Updated last year
- The code enables to perform Bayesian inference in an efficient manner through the use of Hamiltonian Neural Networks (HNNs), Deep Neural …☆14Updated 2 years ago
- AL4PDE: A Benchmark for Active Learning for Neural PDE Solvers☆21Updated last month
- ☆11Updated 4 years ago
- Simulation-based inference in JAX☆31Updated last month
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆10Updated 9 months ago
- A software package for flexible HPC GPs☆16Updated last month
- Kolmogorov-Arnold Networks built on JAX☆68Updated 2 weeks ago
- GATSBI: Generative Adversarial Training for Simulation-Based Inference☆18Updated last year
- ☆10Updated 3 years ago
- Code for Gaussian Score Matching Variational Inference☆33Updated last month
- Public code for running Stochastic Gradient Descent on GPs.☆36Updated 5 months ago
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆24Updated 2 years ago
- An unofficial implementation of the Fourier Neural Operator in Flax☆17Updated 10 months ago
- Implementation of the paper "Neural Wasserstein Gradient Flows for Discrepancies with Riesz Kernels"☆11Updated last year
- ☆19Updated 3 years ago
- Repository for Deterministic Particle Flow Control framework☆10Updated 2 years ago
- Modern normalizing flows in Python. Simple to use and easily extensible.☆11Updated this week
- Matrix-free linear algebra in JAX.☆116Updated 2 months ago
- ☆16Updated 7 months ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago
- Conditional density estimation with neural networks☆30Updated 2 months ago
- Combination of transformers and diffusion models for flexible all-in-one simulation-based inference☆61Updated 9 months ago
- A simple implementation of Hamiltonian Monte Carlo in JAX.☆18Updated last year
- Gradient Based Nested Sampling☆20Updated last year
- Autoencoders in Function Space: Resolution-invariant autoencoders for functional data based on the functional variational autoencoder (FV…☆12Updated 3 months ago
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆46Updated last week
- Semi-supervised Invertible Neural Operators for Bayesian Inverse Problems☆13Updated 10 months ago