NickHunt-Smith / MCMC-diffusion
A Metropolis-Hastings MCMC sampler accelerated via diffusion models
☆13Updated 5 months ago
Alternatives and similar repositories for MCMC-diffusion:
Users that are interested in MCMC-diffusion are comparing it to the libraries listed below
- Efficient Differentiable n-d PDE solvers in JAX.☆22Updated 2 months ago
- Gradient-informed particle MCMC methods☆11Updated 11 months ago
- The code enables to perform Bayesian inference in an efficient manner through the use of Hamiltonian Neural Networks (HNNs), Deep Neural …☆13Updated 2 years ago
- ☆11Updated 3 years ago
- An unofficial implementation of the Fourier Neural Operator in Flax☆17Updated 7 months ago
- GATSBI: Generative Adversarial Training for Simulation-Based Inference☆17Updated last year
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago
- Simulation-based inference in JAX☆29Updated this week
- Codes associated with the manuscript titled "Multi-stage neural networks: Function approximator of machine precision"☆36Updated 9 months ago
- ☆14Updated 5 months ago
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆38Updated 2 months ago
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆49Updated last year
- Physics Informed Neural Networks (PINNs) + SPINNs + HyperPINNs with JAX 📓 Check out our various notebooks to get started ⚠️ Mirror repos…☆16Updated this week
- Modern normalizing flows in Python. Simple to use and easily extensible.☆10Updated 3 weeks ago
- A simple implementation of Hamiltonian Monte Carlo in JAX.☆17Updated 11 months ago
- ☆21Updated last year
- Gradient Based Nested Sampling☆19Updated 11 months ago
- Code for Gaussian Score Matching Variational Inference☆28Updated 3 months ago
- ☆19Updated 2 years ago
- Matrix-free linear algebra in JAX.☆113Updated last week
- ☆37Updated 2 years 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 6 months ago
- Implementation of Action Matching for the Schrödinger equation☆24Updated last year
- Public code for running Stochastic Gradient Descent on GPs.☆36Updated 2 months ago
- Neural Stochastic PDEs: resolution-invariant modelling of continuous spatiotemporal dynamics☆49Updated 2 years ago
- Lightweight MCMC sampling for PyTorch Models aka My Corona Project☆44Updated 4 years ago
- Machine learning assisted marginal likelihood (Bayesian evidence) estimation for Bayesian model selection☆57Updated last month
- PDE-VAE: Variational Autoencoder for Extracting Interpretable Physical Parameters from Spatiotemporal Systems using Unsupervised Learning☆32Updated 3 years ago
- Normalizing Flows with a resampled base distribution☆44Updated 2 years ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆62Updated 6 months ago