Pytorch implementation of "An Optimal Transport Perspective on Unpaired Image Super-Resolution" (JOTA 2025)

PyTorch implementation of "Robust Barycenter Estimation using Semi-unbalanced Neural Optimal Transport" (ICLR 2025)

Tutorial on generative models based on ordinary (ODE) and stochastic (SDE) differential equations

PyTorch implementation of "Light Unbalanced Optimal Transport" (NeurIPS 2024)

Official codebase for "Diffusion Bridge Implicit Models" (ICLR 2025) and "Consistency Diffusion Bridge Models" (NeurIPS 2024)

PyTorch implementation of "Adversarial Schrödinger Bridge Matching" (NeurIPS 2024)

Light and Optimal Schrödinger Bridge Matching (ICML 2024) official PyTorch implementation

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

Pytorch implementation of "Light Schrödinger Bridge" (ICLR 2024)

Pytorch implementation of "Entropic Neural Optimal Transport via Diffusion Processes" (NeurIPS 2023, oral).

Energy-guided Entropic Optimal Transport (ICLR'2024)

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

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

Entropic Optimal Transport Benchmark (NeurIPS 2023).

A research repo for studying different techniques towards making more precise GANs

Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)

A set of tests for evaluating large-scale algorithms for Wasserstein-1 transport computation (NeurIPS'22).

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

PyTorch implementation of Glow