NickHunt-Smith / MCMC-diffusion
A Metropolis-Hastings MCMC sampler accelerated via diffusion models
☆13Updated 6 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
- GATSBI: Generative Adversarial Training for Simulation-Based Inference☆17Updated last year
- 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
- Simulation-based inference in JAX☆30Updated this week
- Efficient Differentiable n-d PDE solvers in JAX.☆24Updated 3 months ago
- Gradient Based Nested Sampling☆19Updated last year
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆41Updated 3 months ago
- Code for Gaussian Score Matching Variational Inference☆32Updated 4 months ago
- Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…☆20Updated 7 months ago
- An unofficial implementation of the Fourier Neural Operator in Flax☆17Updated 8 months ago
- Combination of transformers and diffusion models for flexible all-in-one simulation-based inference☆56Updated 8 months ago
- A simple implementation of Hamiltonian Monte Carlo in JAX.☆18Updated last year
- ☆19Updated 3 years ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago
- Implementation of Action Matching for the Schrödinger equation☆24Updated last year
- Machine learning assisted marginal likelihood (Bayesian evidence) estimation for Bayesian model selection☆60Updated 2 weeks ago
- Repository for Deterministic Particle Flow Control framework☆10Updated 2 years ago
- Conditional density estimation with neural networks☆30Updated last month
- MCHMC: sampler from an arbitrary differentiable distribution☆72Updated 2 weeks ago
- ☆11Updated 3 years ago
- Numerical quadrature with JAX☆53Updated this week
- Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"☆25Updated 9 months ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆65Updated 7 months ago
- Normalizing Flows with a resampled base distribution☆44Updated 2 years ago
- Public code for running Stochastic Gradient Descent on GPs.☆36Updated 3 months ago
- Modular Gaussian Processes☆15Updated 3 years ago
- Kolmogorov-Arnold Networks built on JAX☆56Updated 3 weeks ago
- ☆49Updated last year