Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)
☆42May 8, 2022Updated 3 years ago
Alternatives and similar repositories for Large-Scale-Wasserstein-Gradient-Flows
Users that are interested in Large-Scale-Wasserstein-Gradient-Flows are comparing it to the libraries listed below
Sorting:
- PyTorch implementation of "Light Unbalanced Optimal Transport" (NeurIPS 2024)☆22Dec 23, 2024Updated last year
- PyTorch implementation of "Robust Barycenter Estimation using Semi-unbalanced Neural Optimal Transport" (ICLR 2025)☆17Jul 7, 2025Updated 7 months ago
- [ICML2022] Variational Wasserstein gradient flow☆24Oct 17, 2022Updated 3 years ago
- A set of tests for evaluating large-scale algorithms for Wasserstein-1 transport computation (NeurIPS'22).☆24Sep 9, 2024Updated last year
- PyTorch implementation of "Adversarial Schrödinger Bridge Matching" (NeurIPS 2024)☆28Jun 2, 2025Updated 9 months ago
- Proximal Optimal Transport Modeling of Population Dynamics (AISTATS 2022)☆22Jun 19, 2023Updated 2 years ago
- Pytorch implementation of "Light Schrödinger Bridge" (ICLR 2024)☆64Aug 9, 2024Updated last year
- PyTorch implementation of the paper "Continuous Wasserstein-2 Barycenter Estimation without Minimax Optimization" (ICLR 2021)☆34Jun 17, 2022Updated 3 years ago
- PyTorch implementation of "Kernel Neural Optimal Transport" (ICLR 2023)☆34Oct 3, 2023Updated 2 years ago
- A set of tests for evaluating large-scale algorithms for Wasserstein-2 transport maps computation (NeurIPS 2021)☆47Jun 17, 2022Updated 3 years ago
- Light and Optimal Schrödinger Bridge Matching (ICML 2024) official PyTorch implementation☆59Aug 8, 2024Updated last year
- PyTorch implementation of "Wasserstein Iterative Networks for Barycenter Estimation" (NeurIPS 2022)☆20Jul 3, 2023Updated 2 years ago
- orbital MCMC☆10Jun 17, 2021Updated 4 years ago
- PyTorch implementation of "Extremal Domain Translation with Neural Optimal Transport" (NeurIPS 2023)☆25Dec 24, 2024Updated last year
- Implementation of the paper "Neural Wasserstein Gradient Flows for Discrepancies with Riesz Kernels"☆11Mar 18, 2024Updated last year
- Convex potential flows☆85Nov 19, 2021Updated 4 years ago
- Supervised Training of Conditional Monge Maps☆19Oct 30, 2023Updated 2 years ago
- JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".☆37Mar 18, 2025Updated 11 months ago
- [ICLR 2022] Path integral sampler☆53Aug 21, 2023Updated 2 years ago
- Pytorch implementation of DGflow (ICLR 2021).☆15Mar 3, 2021Updated 5 years ago
- Pytorch implementation of "Entropic Neural Optimal Transport via Diffusion Processes" (NeurIPS 2023, oral).☆43Mar 11, 2024Updated last year
- Entropic Optimal Transport Benchmark (NeurIPS 2023).☆31Mar 21, 2024Updated last year
- ☆10Jun 13, 2022Updated 3 years ago
- Euclidean Wasserstein-2 optimal transportation☆46Aug 19, 2023Updated 2 years ago
- PyTorch Implementation of DSB for Score Based Generative Modeling. Experiments managed using Hydra.☆197Nov 23, 2021Updated 4 years ago
- PyTorch implementation of "Neural Optimal Transport" (ICLR 2023 Spotlight)☆229Dec 11, 2023Updated 2 years ago
- Code of "A Geometric Perspective on Variational Autoencoders" (NeurIPS 2022)☆14Nov 19, 2024Updated last year
- ☆12Jan 9, 2024Updated 2 years ago
- ☆13Mar 26, 2019Updated 6 years ago
- Source code for "Continuous Regularized Wasserstein Barycenters" [NeurIPS 2020].☆15Nov 4, 2020Updated 5 years ago
- ☆51Jul 25, 2024Updated last year
- ☆40Jan 27, 2026Updated last month
- A Metropolis-Hastings MCMC sampler accelerated via diffusion models☆17Jul 25, 2024Updated last year
- ☆11Apr 2, 2021Updated 4 years ago
- PyTorch implementation for "Training and Inference on Any-Order Autoregressive Models the Right Way", NeurIPS 2022 Oral, TPM 2023 Best Pa…☆15May 31, 2023Updated 2 years ago
- A Python library for variational inference with normalizing flow and annealing☆15Mar 13, 2025Updated 11 months ago
- Group Equivariant Linear Maps☆11Jun 4, 2021Updated 4 years ago
- PyTorch implementation of "Wasserstein-2 Generative Networks" (ICLR 2021)☆67Feb 3, 2023Updated 3 years ago
- PyTorch implementation of the OT-Flow approach in arXiv:2006.00104☆57Jul 25, 2024Updated last year