PyTorch implementation of "Neural Optimal Transport" (ICLR 2023 Spotlight)
☆229Dec 11, 2023Updated 2 years ago
Alternatives and similar repositories for NeuralOptimalTransport
Users that are interested in NeuralOptimalTransport are comparing it to the libraries listed below
Sorting:
- PyTorch implementation of "Kernel Neural Optimal Transport" (ICLR 2023)☆34Oct 3, 2023Updated 2 years ago
- PyTorch implementation of "Extremal Domain Translation with Neural Optimal Transport" (NeurIPS 2023)☆25Dec 24, 2024Updated last year
- PyTorch implementation of "Adversarial Schrödinger Bridge Matching" (NeurIPS 2024)☆28Jun 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
- Entropic Optimal Transport Benchmark (NeurIPS 2023).☆31Mar 21, 2024Updated last year
- PyTorch implementation of "Robust Barycenter Estimation using Semi-unbalanced Neural Optimal Transport" (ICLR 2025)☆17Jul 7, 2025Updated 7 months ago
- Official Pytorch implementation of "Neural Optimal Transport with General Cost Functionals" (ICLR 2024)☆24Aug 29, 2024Updated last year
- Pytorch implementation of "Entropic Neural Optimal Transport via Diffusion Processes" (NeurIPS 2023, oral).☆43Mar 11, 2024Updated last year
- PyTorch implementation of "Light Unbalanced Optimal Transport" (NeurIPS 2024)☆22Dec 23, 2024Updated last year
- PyTorch implementation of "Wasserstein Iterative Networks for Barycenter Estimation" (NeurIPS 2022)☆20Jul 3, 2023Updated 2 years ago
- ☆13Sep 13, 2023Updated 2 years ago
- Learning stochastic dynamics from snapshots through regularized unbalanced optimal transport (ICLR'25 oral)☆32Dec 10, 2025Updated 2 months ago
- Pytorch implementation of "Light Schrödinger Bridge" (ICLR 2024)☆64Aug 9, 2024Updated last year
- ☆18Jun 26, 2023Updated 2 years ago
- PyTorch Implementation of Diffusion Schrodinger Bridge Matching☆202Feb 19, 2025Updated last year
- Energy Based Models are a quite novel technique for density estimation. In this university project I explore this new research topic and …☆16Jul 6, 2021Updated 4 years ago
- The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)☆110Dec 19, 2024Updated last year
- A Pytorch implementation of the optimal transport kernel embedding☆119Feb 10, 2021Updated 5 years ago
- Implementation of the Gromov-Wasserstein distance to the setting of Unbalanced Optimal Transport☆45Mar 13, 2023Updated 2 years ago
- A list of awesome papers and cool resources on optimal transport and its applications in general! As you will notice, this list is curren…☆246May 9, 2021Updated 4 years ago
- PyTorch Implementation of DSB for Score Based Generative Modeling. Experiments managed using Hydra.☆197Nov 23, 2021Updated 4 years ago
- This is the code for the paper Embrace the Gap: VAEs perform Independent Mechanism Analysis, showing that optimizing the ELBO is equivale…☆23Apr 22, 2024Updated last year
- Python package for P2 (Path Planning), a masked diffusion model sampling method for sequence generation (protein, text, etc.).☆23Aug 19, 2025Updated 6 months ago
- Simplified Diffusion Schrödinger Bridge☆13Apr 19, 2024Updated last year
- Lagrangian formulation of Doob's h-transform allowing for efficient rare event sampling☆57Mar 26, 2025Updated 11 months ago
- ☆18Oct 6, 2024Updated last year
- Python package for the ICML 2022 paper "Unsupervised Ground Metric Learning Using Wasserstein Singular Vectors".☆10Sep 2, 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
- Neural Optimal Transport with Lagrangian Costs☆61May 20, 2025Updated 9 months ago
- [ICML 2022] Representation Topology Divergence: A Method for Comparing Neural Network Representations☆26Aug 15, 2022Updated 3 years ago
- [ICML2023] InfoOT: Information Maximizing Optimal Transport☆41Apr 27, 2023Updated 2 years ago
- Implementation of our ICLR2023 paper "Spherical-Sliced Wasserstein"☆14Feb 24, 2026Updated last week
- Tutorial on generative models based on ordinary (ODE) and stochastic (SDE) differential equations☆46Nov 14, 2023Updated 2 years ago
- implements optimal transport algorithms in pytorch☆105May 15, 2022Updated 3 years ago
- PyTorch implementation of the paper "Continuous Wasserstein-2 Barycenter Estimation without Minimax Optimization" (ICLR 2021)☆34Jun 17, 2022Updated 3 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
- A set of tests for evaluating large-scale algorithms for Wasserstein-2 transport maps computation (NeurIPS 2021)☆47Jun 17, 2022Updated 3 years ago
- POT : Python Optimal Transport☆2,763Feb 24, 2026Updated last week