NickHunt-Smith / MCMC-diffusion
A Metropolis-Hastings MCMC sampler accelerated via diffusion models
☆13Updated 7 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
- 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
- ☆11Updated 3 years ago
- A software package for flexible HPC GPs☆16Updated 3 weeks ago
- Gradient-informed particle MCMC methods☆11Updated last year
- Machine learning assisted marginal likelihood (Bayesian evidence) estimation for Bayesian model selection☆60Updated last month
- Gradient Based Nested Sampling☆20Updated last year
- Simulation-based inference in JAX☆31Updated last month
- ☆19Updated 3 years ago
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆49Updated last year
- Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"☆24Updated 10 months ago
- Kolmogorov-Arnold Networks built on JAX☆64Updated this week
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆37Updated 2 years ago
- An unofficial implementation of the Fourier Neural Operator in Flax☆17Updated 9 months ago
- Codes associated with the manuscript titled "Multi-stage neural networks: Function approximator of machine precision"☆38Updated 11 months ago
- Code for Gaussian Score Matching Variational Inference☆32Updated last month
- Combination of transformers and diffusion models for flexible all-in-one simulation-based inference☆60Updated 9 months ago
- Public code for running Stochastic Gradient Descent on GPs.☆36Updated 4 months ago
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆97Updated 11 months ago
- Matrix-free linear algebra in JAX.☆116Updated 2 months ago
- Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax☆28Updated 10 months ago
- Normalizing Flows with a resampled base distribution☆45Updated 2 years ago
- Numerical quadrature with JAX☆53Updated this week
- Modern normalizing flows in Python. Simple to use and easily extensible.☆11Updated last week
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆66Updated this week
- AL4PDE: A Benchmark for Active Learning for Neural PDE Solvers☆19Updated 3 weeks ago
- Conditional density estimation with neural networks☆30Updated 2 months ago