iamalexkorotin / NeuralOptimalTransportView external linksLinks
PyTorch implementation of "Neural Optimal Transport" (ICLR 2023 Spotlight)
☆226Dec 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)☆29Oct 3, 2023Updated 2 years ago
- PyTorch implementation of "Extremal Domain Translation with Neural Optimal Transport" (NeurIPS 2023)☆21Dec 24, 2024Updated last year
- Entropic Optimal Transport Benchmark (NeurIPS 2023).☆27Mar 21, 2024Updated last year
- PyTorch implementation of "Robust Barycenter Estimation using Semi-unbalanced Neural Optimal Transport" (ICLR 2025)☆17Jul 7, 2025Updated 7 months ago
- Pytorch implementation of "Entropic Neural Optimal Transport via Diffusion Processes" (NeurIPS 2023, oral).☆42Mar 11, 2024Updated last year
- PyTorch implementation of "Light Unbalanced Optimal Transport" (NeurIPS 2024)☆22Dec 23, 2024Updated last year
- Pytorch implementation of "Neural Optimal Transport with General Cost Functionals" (ICLR 2024)☆23Aug 29, 2024Updated last year
- Light and Optimal Schrödinger Bridge Matching (ICML 2024) official PyTorch implementation☆57Aug 8, 2024Updated last year
- ☆13Sep 13, 2023Updated 2 years ago
- Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)☆42May 8, 2022Updated 3 years ago
- Learning stochastic dynamics from snapshots through regularized unbalanced optimal transport (ICLR'25 oral)☆31Dec 10, 2025Updated 2 months ago
- Generative Modeling with Optimal Transport Maps - ICLR 2022☆60May 3, 2022Updated 3 years ago
- ☆22Feb 4, 2025Updated last year
- ☆18Jun 26, 2023Updated 2 years ago
- 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
- A Pytorch implementation of the optimal transport kernel embedding☆119Feb 10, 2021Updated 5 years ago
- Code for "HyperSDFusion: Bridging Hierarchical Structures in Language and Geometry for Enhanced 3D Text2Shape Generation" CVPR2024☆10Apr 19, 2024Updated last year
- Implementation of the Gromov-Wasserstein distance to the setting of Unbalanced Optimal Transport☆45Mar 13, 2023Updated 2 years ago
- PyTorch Implementation of DSB for Score Based Generative Modeling. Experiments managed using Hydra.☆191Nov 23, 2021Updated 4 years ago
- ☆16Oct 6, 2024Updated last year
- Python package for P2 (Path Planning), a masked diffusion model sampling method for sequence generation (protein, text, etc.).☆23Aug 19, 2025Updated 5 months ago
- Lagrangian formulation of Doob's h-transform allowing for efficient rare event sampling☆55Mar 26, 2025Updated 10 months ago
- The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)☆110Dec 19, 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
- [ICML 2022] Representation Topology Divergence: A Method for Comparing Neural Network Representations☆25Aug 15, 2022Updated 3 years ago
- [ICML2023] InfoOT: Information Maximizing Optimal Transport☆41Apr 27, 2023Updated 2 years ago
- Tutorial on generative models based on ordinary (ODE) and stochastic (SDE) differential equations☆47Nov 14, 2023Updated 2 years ago
- PyTorch implementation of "Wasserstein-2 Generative Networks" (ICLR 2021)☆66Feb 3, 2023Updated 3 years ago
- PyTorch implementation of the paper "Continuous Wasserstein-2 Barycenter Estimation without Minimax Optimization" (ICLR 2021)☆33Jun 17, 2022Updated 3 years ago
- A minimal, single-file implementation of the MeanFlow paper on 2D toy examples, with a side-by-side comparison to rectified flow.☆27Jul 4, 2025Updated 7 months ago
- Graph matching and clustering by comparing heat kernels via optimal transport.☆27Dec 8, 2022Updated 3 years ago
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆70Mar 4, 2025Updated 11 months ago
- ☆14May 17, 2023Updated 2 years ago
- Supervised Training of Conditional Monge Maps☆19Oct 30, 2023Updated 2 years ago
- Deep Generative Models course, 2023☆62Apr 13, 2024Updated last year
- ☆39Jan 27, 2026Updated 2 weeks ago
- Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.☆700Feb 5, 2026Updated last week
- Deep Generative Models course, 2022-2023☆69Jul 19, 2023Updated 2 years ago
- TorchCFM: a Conditional Flow Matching library☆2,294Nov 11, 2025Updated 3 months ago