NickHunt-Smith / MCMC-diffusion
A Metropolis-Hastings MCMC sampler accelerated via diffusion models
☆13Updated 9 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
- Gradient-informed particle MCMC methods☆11Updated last year
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆11Updated 10 months ago
- Efficient Differentiable n-d PDE solvers in JAX.☆28Updated 5 months ago
- 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
- GATSBI: Generative Adversarial Training for Simulation-Based Inference☆18Updated last year
- AL4PDE: A Benchmark for Active Learning for Neural PDE Solvers☆22Updated last month
- Simulation-based inference in JAX☆31Updated 2 weeks ago
- Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"☆25Updated last year
- Hamiltonian Neural Networks for solving Differential Equations☆22Updated 3 years ago
- Gradient Based Nested Sampling☆20Updated last year
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆46Updated 3 weeks ago
- ☆10Updated 3 years ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆67Updated this week
- An unofficial implementation of the Fourier Neural Operator in Flax☆17Updated 11 months ago
- Modern normalizing flows in Python. Simple to use and easily extensible.☆11Updated 3 weeks ago
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆37Updated 2 years ago
- Conditional density estimation with neural networks☆30Updated 3 months ago
- A simple implementation of Hamiltonian Monte Carlo in JAX.☆18Updated last year
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆103Updated last year
- Combination of transformers and diffusion models for flexible all-in-one simulation-based inference☆63Updated 10 months ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago
- Probabilistic modeling of tabular data with normalizing flows.☆56Updated 3 weeks ago
- Kolmogorov-Arnold Networks built on JAX☆74Updated 2 weeks ago
- MCHMC: sampler from an arbitrary differentiable distribution☆72Updated last month
- Robust initialisation of inducing points in sparse variational GP regression models.☆33Updated 2 years ago
- Code for Gaussian Score Matching Variational Inference☆33Updated 2 months ago
- Machine learning assisted marginal likelihood (Bayesian evidence) estimation for Bayesian model selection☆64Updated 2 weeks ago
- ☆11Updated 4 years ago
- Codes associated with the manuscript titled "Multi-stage neural networks: Function approximator of machine precision"☆40Updated last year