kachayev

PSWE

Implementation of Pooling by Sliced-Wasserstein Embedding (NeurIPS 2021)

OTK

A Pytorch implementation of the optimal transport kernel embedding

ott

Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.

Learning-Embeddings-into-Entropic-Wasserstein-Spaces-ENSAE

A thorough review of the paper "Learning Embeddings into Entropic Wasserstein Spaces" by Frogner et al. Includes a reproduction of the results on word embeddings.

S-WMD

Code for Supervised Word Mover's Distance (SWMD)

swav

PyTorch implementation of SwAV https//arxiv.org/abs/2006.09882

backpacks-flash-attn

The original Backpack Language Model implementation, a fork of FlashAttention

gwil

Cross-Domain Imitation Learning via Optimal Transport

Relational-AutoEncoders

Learning Autoencoders with Relational Regularization

Graph-Optimal-Transport

Code for ICML 2020 "Graph Optimal Transport for Cross-Domain Alignment"

fugw

Scalable python GPU solvers for fused unbalanced gromov-wasserstein optimal transport problems, with routines and examples to align brain data (fMRI)

CEBRA

Learnable latent embeddings for joint behavioral and neural analysis - Official implementation of CEBRA

PPMM

Python3 implementation of the paper [Large-scale optimal transport map estimation using projection pursuit]

Large-Scale-OT

Stochastic algorithms for computing Regularized Optimal Transport

MissingDataOT

A Pytorch implementation of missing data imputation using optimal transport.

COOT

CO-Optimal Transport

learning-to-generate-wasserstein-barycenters

relational-networks

Pytorch implementation of "A simple neural network module for relational reasoning" (Relational Networks)

RelationNetworks-CLEVR

A pytorch implementation for "A simple neural network module for relational reasoning", working on the CLEVR dataset

StochasticOT

Stochastic Optimization for Optimal Transport

demd

Efficient Computation of d-Dimensional Earth Mover's Distance

POT

POT : Python Optimal Transport

gsw

DSW

Distributional Sliced-Wasserstein distance code

s-gwl

Scalable Gromov-Wasserstein Learning for Graph Partitioning and Matching

Spectral-Gromov-Wasserstein

Graph matching and clustering by comparing heat kernels via optimal transport.

2016-ICML-gromov-wasserstein

Gabriel Peyré, Marco Cuturi, Justin Solomon, Gromov-Wasserstein Averaging of Kernel and Distance Matrices, Proc. of ICML 2016.

sinkformers

Transformers with doubly stochastic attention

mazelab

A customizable framework to create maze and gridworld environments

WassersteinDictionaryLearning

Morgan A. Schmitz., Matthieu Heitz, Nicolas Bonneel, Fred Ngole, David Coeurjolly, Marco Cuturi, Gabriel Peyré, and Jean-Luc Starck. "Wasserstein dictionary learning: Optimal transport-based unsupervised nonlinear dictionary learning." SIAM Journal on Imaging Sciences, 2018