PyTorch implementation of "Wasserstein Iterative Networks for Barycenter Estimation" (NeurIPS 2022)
☆20Jul 3, 2023Updated 2 years ago
Alternatives and similar repositories for WassersteinIterativeNetworks
Users that are interested in WassersteinIterativeNetworks are comparing it to the libraries listed below
Sorting:
- PyTorch implementation of "Robust Barycenter Estimation using Semi-unbalanced Neural Optimal Transport" (ICLR 2025)☆17Jul 7, 2025Updated 8 months ago
- PyTorch implementation of "Extremal Domain Translation with Neural Optimal Transport" (NeurIPS 2023)☆25Dec 24, 2024Updated last year
- PyTorch implementation of "Light Unbalanced Optimal Transport" (NeurIPS 2024)☆22Dec 23, 2024Updated last year
- PyTorch implementation of the paper "Continuous Wasserstein-2 Barycenter Estimation without Minimax Optimization" (ICLR 2021)☆34Jun 17, 2022Updated 3 years ago
- Entropic Optimal Transport Benchmark (NeurIPS 2023).☆32Mar 21, 2024Updated last year
- PyTorch implementation of "Kernel Neural Optimal Transport" (ICLR 2023)☆34Oct 3, 2023Updated 2 years ago
- PyTorch implementation of "Adversarial Schrödinger Bridge Matching" (NeurIPS 2024)☆29Jun 2, 2025Updated 9 months 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 "Entropic Neural Optimal Transport via Diffusion Processes" (NeurIPS 2023, oral).☆43Mar 11, 2024Updated 2 years ago
- Light and Optimal Schrödinger Bridge Matching (ICML 2024) official PyTorch implementation☆59Aug 8, 2024Updated last year
- Python package for the ICML 2022 paper "Unsupervised Ground Metric Learning Using Wasserstein Singular Vectors".☆10Sep 2, 2024Updated last year
- Pytorch implementation of "Light Schrödinger Bridge" (ICLR 2024)☆65Aug 9, 2024Updated last year
- Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)☆42May 8, 2022Updated 3 years ago
- Source code for "Continuous Regularized Wasserstein Barycenters" [NeurIPS 2020].☆15Nov 4, 2020Updated 5 years ago
- Tree-Based Diffusion Schrödinger Bridge with Applications to Wasserstein Barycenters☆10Mar 5, 2024Updated 2 years ago
- [NeurIPS 2021] Manifold Topology Divergence: a Framework for Comparing Data Manifolds☆15Mar 1, 2022Updated 4 years ago
- PyTorch implementation of "Neural Optimal Transport" (ICLR 2023 Spotlight)☆228Dec 11, 2023Updated 2 years ago
- conditional U-Net Pytorch Implementation☆11Jul 27, 2021Updated 4 years ago
- Multimodal deep learning in neuroimaging☆14Jan 27, 2023Updated 3 years ago
- We investigated corruption robustness across different architectures including Convolutional Neural Networks, Vision Transformers, and th…☆16Oct 28, 2021Updated 4 years ago
- ☆12Oct 14, 2018Updated 7 years ago
- This repository contains the code for our ECCV 2022 paper on our "Non-isotropic Probabilistic Take on Proxy-based Deep Metric Learning".☆12Dec 6, 2022Updated 3 years ago
- This package is dedicated to high-order optimization methods. All the methods can be used similarly to standard PyTorch optimizers.☆29Jun 17, 2025Updated 9 months ago
- A set of tests for evaluating large-scale algorithms for Wasserstein-2 transport maps computation (NeurIPS 2021)☆47Jun 17, 2022Updated 3 years ago
- Code and raw result files of our CVPR2020 oral paper "Recursive Least-Squares Estimator-Aided Online Learning for Visual Tracking""☆18Dec 4, 2020Updated 5 years ago
- ☆30Jul 4, 2025Updated 8 months ago
- Importance of Self-Consistency in Active Learning for Semantic Segmentation (BMVC 2020)☆17Oct 11, 2021Updated 4 years ago
- This repository contains the code for our CVPR 2022 paper on "Non-isotropy Regularization for Proxy-based Deep Metric Learning".☆15Mar 10, 2023Updated 3 years ago
- Visual Concept Connectome☆15Jun 23, 2024Updated last year
- Code for our ICLR19 paper "Wasserstein Barycenters for Model Ensembling", Pierre Dognin, Igor Melnyk, Youssef Mroueh, Jarret Ross, Cicero…☆23Nov 18, 2019Updated 6 years ago
- Official Pytorch implementation of "Neural Optimal Transport with General Cost Functionals" (ICLR 2024)☆24Aug 29, 2024Updated last year
- Implementation of WGAN-QC☆16Nov 25, 2019Updated 6 years ago
- PyTorch implementation of "Wasserstein-2 Generative Networks" (ICLR 2021)☆67Feb 3, 2023Updated 3 years ago
- Sinkhorn Barycenters via Frank-Wolfe algorithm☆27Feb 3, 2020Updated 6 years ago
- Open-source framework for science presentations and talks, generating interactive HTML from simple Python based interface.☆13Jul 21, 2025Updated 7 months ago
- machine learning models for predicting depression based on EEG data☆24Sep 9, 2022Updated 3 years ago
- Generative Modeling with Optimal Transport Maps - ICLR 2022☆60May 3, 2022Updated 3 years ago
- Official code implementation for the paper "Do Vision & Language Decoders use Images and Text equally? How Self-consistent are their Expl…☆12Apr 4, 2025Updated 11 months ago
- ☆16Apr 28, 2023Updated 2 years ago