modichirag / GSM-VILinks
Code for Gaussian Score Matching Variational Inference
☆35Updated 9 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
- Simulation-based inference in JAX☆41Updated 2 weeks ago
- Simulation-based inference benchmark☆106Updated 10 months ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆105Updated 2 years ago
- Matrix-free linear algebra in JAX.☆151Updated last week
- Normalizing Flows using JAX☆85Updated 2 years ago
- ☆30Updated 2 years ago
- Likelihood-free AMortized Posterior Estimation with PyTorch☆131Updated last year
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆12Updated last year
- Public code for running Stochastic Gradient Descent on GPs.☆39Updated 7 months ago
- Implementation of Action Matching for the Schrödinger equation☆25Updated 2 years ago
- A generic interface for linear algebra backends☆75Updated 3 weeks ago
- Combination of transformers and diffusion models for flexible all-in-one simulation-based inference☆75Updated 5 months ago
- Generative models with JAX.☆20Updated 11 months ago
- Normalizing Flows with a resampled base distribution☆47Updated 3 years ago
- MCHMC: sampler from an arbitrary differentiable distribution☆75Updated this week
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆50Updated 4 months ago
- Stainless neural networks in JAX☆34Updated 3 months ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆117Updated last month
- ☆203Updated last month
- Riemannian Optimization Using JAX☆53Updated 2 years ago
- Code for efficiently sampling functions from GP(flow) posteriors☆74Updated 5 years ago
- Sliced Iterative Generator (SIG) & Gaussianizing Iterative Slicing (GIS)☆37Updated 2 years ago
- Lightweight MCMC sampling for PyTorch Models aka My Corona Project☆51Updated 4 months ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 4 years ago
- Stochastic Normalizing Flows☆78Updated 4 years ago
- A collection of graph neural networks implementations in JAX☆35Updated 2 years ago
- Probabilistic modeling of tabular data with normalizing flows.☆58Updated this week
- A simple implementation of Hamiltonian Monte Carlo in JAX.☆21Updated last year
- Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)☆39Updated 3 years ago