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
- Efficient Differentiable n-d PDE solvers in JAX.☆40Updated 7 months ago
- 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
- An unofficial implementation of the Fourier Neural Operator in Flax☆17Updated last year
- Autoencoders in Function Space: Resolution-invariant autoencoders for functional data based on the functional variational autoencoder (FV…☆14Updated 6 months ago
- Gradient-informed particle MCMC methods☆12Updated last year
- Simulation-based inference in JAX☆32Updated 2 months ago
- Implementation of the paper "Neural Wasserstein Gradient Flows for Discrepancies with Riesz Kernels"☆11Updated last year
- Code for Gaussian Score Matching Variational Inference☆34Updated 4 months ago
- ☆11Updated 4 years ago
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆49Updated 2 months ago
- Implementation of Action Matching for the Schrödinger equation☆24Updated 2 years ago
- Differentiable and accelerated wavelet transform on the sphere with JAX☆20Updated last month
- Kolmogorov-Arnold Networks built on JAX☆82Updated 2 months ago
- Gradient Based Nested Sampling☆20Updated last year
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆11Updated last year
- Normalizing Flows with a resampled base distribution☆47Updated 2 years ago
- GATSBI: Generative Adversarial Training for Simulation-Based Inference☆18Updated last year
- Conditional density estimation with neural networks☆31Updated 5 months ago
- Conditional Flow Matching using JAX☆13Updated 2 years ago
- A software package for flexible HPC GPs☆16Updated this week
- ☆19Updated 3 years ago
- Matrix-free linear algebra in JAX.☆125Updated last month
- Gaussian Mixture Models in Jax☆20Updated 4 months ago
- Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"☆30Updated last year
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆73Updated 2 months ago
- Collecting research materials on neural samplers with diffusion/flow models☆52Updated 3 weeks ago
- Code for "Log Neural Controlled Differential Equations: The Lie Brackets Make a Difference" (ICML 2024)☆20Updated last month
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆38Updated 2 years ago