modichirag / GSM-VILinks
Code for Gaussian Score Matching Variational Inference
☆35Updated 10 months ago
Alternatives and similar repositories for GSM-VI
Users that are interested in GSM-VI are comparing it to the libraries listed below
Sorting:
- ☆52Updated 2 years ago
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆13Updated last year
- Simulation-based inference in JAX☆43Updated last month
- Simulation-based inference benchmark☆108Updated 11 months ago
- Likelihood-free AMortized Posterior Estimation with PyTorch☆132Updated last year
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆105Updated 2 years ago
- Implementation of Action Matching for the Schrödinger equation☆25Updated 2 years ago
- Normalizing Flows using JAX☆86Updated 2 years ago
- ☆30Updated 2 years ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 4 years ago
- Tutorial materials of the Probabilistic Numerics Spring School.☆35Updated 2 years ago
- Lightweight MCMC sampling for PyTorch Models aka My Corona Project☆52Updated 5 months ago
- Combination of transformers and diffusion models for flexible all-in-one simulation-based inference☆79Updated 7 months ago
- Matrix-free linear algebra in JAX.☆160Updated last month
- Conditional Flow Matching using JAX☆13Updated 2 years ago
- Stainless neural networks in JAX☆34Updated last month
- Generative models with JAX.☆20Updated last year
- A generic interface for linear algebra backends☆76Updated 2 months ago
- Sliced Iterative Generator (SIG) & Gaussianizing Iterative Slicing (GIS)☆37Updated 2 years ago
- ☆209Updated this week
- MCHMC: sampler from an arbitrary differentiable distribution☆75Updated last month
- Code for efficiently sampling functions from GP(flow) posteriors☆74Updated 5 years ago
- Normalizing Flows with a resampled base distribution☆47Updated 3 years ago
- Community-sourced list of papers and resources on neural simulation-based inference.☆139Updated 2 weeks ago
- Public code for running Stochastic Gradient Descent on GPs.☆39Updated 8 months ago
- Riemannian Optimization Using JAX☆53Updated 2 years ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆123Updated last month
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆255Updated 2 months ago
- Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)☆40Updated 3 years ago
- A collection of graph neural networks implementations in JAX☆35Updated 2 years ago