iamalexkorotin

Alexander Korotin's repositories

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

Language:Jupyter NotebookMIT162 4 6

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

Language:Jupyter NotebookMIT49 5 1

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

Language:Jupyter NotebookMIT34 3 1

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

Language:Jupyter NotebookMIT21 10

PyTorch implementation of the paper "Continuous Wasserstein-2 Barycenter Estimation without Minimax Optimization" (ICLR 2021)

Language:Jupyter NotebookMIT21 10

PyTorch implementation of "Wasserstein Iterative Networks for Barycenter Estimation" (NeurIPS 2022)

Language:Jupyter NotebookMIT9 3 2

Entropic Optimal Transport Benchmark (NeurIPS 2023)

Language:Jupyter NotebookMIT100

Pytorch implementation of "Light Schrödinger Bridge".

Language:PythonMIT100

PyTorch source code of Generative Modeling with Optimal Transport Maps (ICLR 2022).

Language:Jupyter Notebook000

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

Language:Jupyter NotebookMIT000

A set of tests for evaluating large-scale algorithms for Wasserstein-1 transport computation.

Language:Jupyter NotebookMIT000

Energy-guided Entropic Optimal Transport (ICLR'2024)

MIT000

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

MIT000

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

000

Light and Optimal Schrödinger Bridge Matching official PyTorch implementation

MIT000