NickHunt-Smith / MCMC-diffusionLinks
A Metropolis-Hastings MCMC sampler accelerated via diffusion models
☆13Updated 11 months ago
Alternatives and similar repositories for MCMC-diffusion
Users that are interested in MCMC-diffusion are comparing it to the libraries listed below
Sorting:
- ☆27Updated last year
- The code enables to perform Bayesian inference in an efficient manner through the use of Hamiltonian Neural Networks (HNNs), Deep Neural …☆15Updated 2 years ago
- Autoencoders in Function Space: Resolution-invariant autoencoders for functional data based on the functional variational autoencoder (FV…☆15Updated 7 months ago
- Conditional density estimation with neural networks☆31Updated 5 months ago
- Simulation-based inference in JAX☆32Updated 2 months ago
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆11Updated last year
- Code for "Log Neural Controlled Differential Equations: The Lie Brackets Make a Difference" (ICML 2024)☆20Updated last month
- Gradient-informed particle MCMC methods☆12Updated last year
- Efficient Differentiable n-d PDE solvers in JAX.☆45Updated 8 months ago
- An unofficial implementation of the Fourier Neural Operator in Flax☆17Updated last year
- GATSBI: Generative Adversarial Training for Simulation-Based Inference☆18Updated last year
- Conditional Flow Matching using JAX☆13Updated 2 years ago
- ☆19Updated 3 years ago
- Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)☆33Updated 3 years ago
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆49Updated 3 months ago
- Combination of transformers and diffusion models for flexible all-in-one simulation-based inference☆68Updated 2 weeks ago
- Kolmogorov-Arnold Networks built on JAX☆82Updated 2 months ago
- Differentiable and accelerated wavelet transform on the sphere with JAX☆20Updated last month
- Code for Gaussian Score Matching Variational Inference☆34Updated 4 months ago
- ☆11Updated 4 years ago
- Machine learning assisted marginal likelihood (Bayesian evidence) estimation for Bayesian model selection☆69Updated last month
- Implementation of the paper "Neural Wasserstein Gradient Flows for Discrepancies with Riesz Kernels"☆11Updated last year
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆74Updated 2 months ago
- Code for the paper "Rational neural networks", NeurIPS 2020☆28Updated 4 years ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago
- Hamiltonian Neural Networks for solving Differential Equations☆22Updated 3 years ago
- Normalizing Flows with a resampled base distribution☆47Updated 2 years ago
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆52Updated 2 years ago
- [ICML2022] Variational Wasserstein gradient flow☆23Updated 2 years ago
- A software package for flexible HPC GPs☆16Updated this week