marc-h-lambert / W-VI
The companion code for the paper "Variational inference via Wasserstein gradient flows (W-VI) M. Lambert, S. Chewi, F. Bach, S. Bonnabel, P. Rigollet."
☆11Updated 2 years ago
Alternatives and similar repositories for W-VI:
Users that are interested in W-VI are comparing it to the libraries listed below
- ☆32Updated 2 years ago
- Mutual information estimators and benchmark☆44Updated last month
- Demos for the paper Generalized Variational Inference (Knoblauch, Jewson & Damoulas, 2019)☆20Updated 5 years ago
- Repository for the paper "Riemannian Laplace approximations for Bayesian neural networks"☆11Updated last year
- Code for ICE-BeeM paper - NeurIPS 2020☆86Updated 3 years ago
- Official code for the ICLR 2021 paper Neural ODE Processes☆70Updated 2 years ago
- All You Need is a Good Functional Prior for Bayesian Deep Learning (JMLR 2022)☆21Updated 2 years ago
- Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)☆32Updated 2 years ago
- The Wasserstein Distance and Optimal Transport Map of Gaussian Processes☆52Updated 4 years ago
- Simple (and cheap!) neural network uncertainty estimation☆61Updated 3 weeks ago
- Code for 'Memory-based dual Gaussian processes for sequential learning' (ICML 2023)☆12Updated last year
- Deep universal probabilistic programming with Python and PyTorch☆11Updated 4 years ago
- [NeurIPS 2023] Riemannian Residual Neural Networks (https://arxiv.org/abs/2006.10254)☆18Updated last year
- ☆23Updated last year
- Learning the optimal transport map via input convex neural neworks☆41Updated 4 years ago
- Python notebooks for Optimal Transport between Gaussian Mixture Models☆41Updated 3 years ago
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆97Updated 10 months ago
- Code for the paper "Disentanglement by Nonlinear ICA with General Incompressible-flow Networks (GIN)" (2020)☆32Updated 3 years ago
- This repository contains code for applying Riemannian geometry in machine learning.☆78Updated 3 years ago
- On Uncertainty, Tempering, and Data Augmentation in Bayesian Classification☆21Updated 2 years ago
- Pytorch implementation of "Entropic Neural Optimal Transport via Diffusion Processes" (NeurIPS 2023, oral).☆36Updated 11 months ago
- Algorithms for computations on random manifolds made easier☆87Updated last year
- ☆16Updated 2 years ago
- Bayesian active learning with EPIG data acquisition☆28Updated last month
- ☆12Updated 7 months ago
- Implementation of a PyTorch Mutual Information Estimation Toolkit☆15Updated 11 months ago
- VAEs and nonlinear ICA: a unifying framework☆33Updated 4 years ago
- Official Implementation of the paper: "A Rate-Distorion View of Uncertainty Quantification", ICML 2024☆28Updated 6 months ago
- PyTorch implementation of the OT-Flow approach in arXiv:2006.00104☆51Updated 7 months ago
- Repo for our paper "Repulsive deep ensembles are Bayesian"☆19Updated 3 years ago