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
- Repository for the paper "Riemannian Laplace approximations for Bayesian neural networks"☆11Updated last year
- Demos for the paper Generalized Variational Inference (Knoblauch, Jewson & Damoulas, 2019)☆20Updated 6 years ago
- Official code for the ICLR 2021 paper Neural ODE Processes☆72Updated 2 years ago
- Code for 'Memory-based dual Gaussian processes for sequential learning' (ICML 2023)☆12Updated last year
- 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
- Learning the optimal transport map via input convex neural neworks☆41Updated 4 years ago
- Pytorch implementation of "Entropic Neural Optimal Transport via Diffusion Processes" (NeurIPS 2023, oral).☆37Updated last year
- The Wasserstein Distance and Optimal Transport Map of Gaussian Processes☆52Updated 4 years ago
- Lightweight MCMC sampling for PyTorch Models aka My Corona Project☆45Updated 4 years ago
- Repo for our paper "Repulsive deep ensembles are Bayesian"☆19Updated 3 years ago
- Implementation of the Gromov-Wasserstein distance to the setting of Unbalanced Optimal Transport☆44Updated 2 years ago
- ☆13Updated last year
- Methods and experiments for assumed density SDE approximations☆11Updated 3 years ago
- Python notebooks for Optimal Transport between Gaussian Mixture Models☆45Updated 3 years ago
- [NeurIPS 2023] Riemannian Residual Neural Networks (https://arxiv.org/abs/2006.10254)☆19Updated last year
- Code to accompany paper 'Bayesian Deep Ensembles via the Neural Tangent Kernel'☆26Updated 4 years ago
- Official Implementation of the paper: "A Rate-Distorion View of Uncertainty Quantification", ICML 2024☆28Updated 7 months ago
- Deep universal probabilistic programming with Python and PyTorch☆12Updated 5 years ago
- Simple (and cheap!) neural network uncertainty estimation☆63Updated this week
- Mutual information estimators and benchmark☆47Updated 2 months ago
- Repository for the work Transforming Gaussian Processes with Normalizing Flows published at AISTATS 2021☆24Updated 2 years ago
- PyTorch implementation of the paper "Continuous Wasserstein-2 Barycenter Estimation without Minimax Optimization" (ICLR 2021)☆23Updated 2 years ago
- Tensorflow implementation for the SVGP-VAE model.☆22Updated 3 years ago
- ☆23Updated 2 years ago
- Code for the paper "Disentanglement by Nonlinear ICA with General Incompressible-flow Networks (GIN)" (2020)☆32Updated 3 years ago
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆97Updated 11 months ago
- ☆32Updated 2 years ago
- Image-to-image regression with uncertainty quantification in PyTorch. Take any dataset and train a model to regress images to images with…☆55Updated 2 years ago
- Code for the paper "Beyond Deep Ensembles: A Large-Scale Evaluation of Bayesian Deep Learning under Distribution Shift"☆22Updated last year