Popular GitHub Repositories for Optimal Transport

TorchCFM: a Conditional Flow Matching library

Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.

[NeurIPS 2024] GaussianCube: A Structured and Explicit Radiance Representation for 3D Generative Modeling

PyTorch implementation of "Neural Optimal Transport" (ICLR 2023 Spotlight)

[ICLR2023] PLOT: Prompt Learning with Optimal Transport for Vision-Language Models

Neural Architecture Search with Bayesian Optimisation and Optimal Transport

Implementation of the Sliced Wasserstein Autoencoder using PyTorch

PyTorch implementation of slicing adversarial network (SAN)

The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)

PyTorch implementation of "Wasserstein-2 Generative Networks" (ICLR 2021)

A set of tests for evaluating large-scale algorithms for Wasserstein-2 transport maps computation (NeurIPS 2021)

Unsupervised Learning of Graph Hierarchical Abstractions with Differentiable Coarsening and Optimal Transport

A PyTorch implementation of adaptive Monte Carlo Optimal Transport algorithm

Unofficial PyTorch implementation of "GAN-QP: A Novel GAN Framework without Gradient Vanishing and Lipschitz Constraint"

The official code repo for the paper "Mixture of Stochastic Experts for Modeling Aleatoric Uncertainty in Segmentation". (ICLR 2023)

Official repository of the "Transformer Fusion with Optimal Transport" paper, published as a conference paper at ICLR 2024.

PyTorch implementation of "Kernel Neural Optimal Transport" (ICLR 2023)

Multi-Source Domain Adaptation via Optimal Transport for Student-Teacher Learning - UAI 2021

PyTorch implementation of "Extremal Domain Translation with Neural Optimal Transport" (NeurIPS 2023)

Source code for "Training Generative Adversarial Networks Via Turing Test".