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

Implementation of DETR in JAX

Code for the paper Geodesic Sinkhorn for Fast and Accurate Optimal Transport on Manifolds.

An easy-to-use Python library for processing and manipulating 3D point clouds and meshes.

[ICML 2023] Official code for our paper: 'Conditional Tree Matching for Inference-Time Adaptation of Tree Prediction Models'

A neural parser for typelogical grammars based on Sinkhorn networks and Linear Logic Proof Nets.

Library for solving variational mean-field games using optimal transport and the Sinkhorn algorithm

Implementation of the paper "Anticipation-Free Training for Simultaneous Machine Translation"

Douglas-Rachford Splitting for Optimal Transport

Code for ECML/PKDD paper: "LSMI-Sinkhorn: Semi-supervised Mutual Information Estimation with Optimal Transport"

A dedicated convenient repo for different Music Transformers implementations (Reformer/XTransformer/Sinkhorn/etc)

Code for some of the algorithms used in Optimal Transport like subspace alignment and sinkhorn algorithms

Code for "A MFG Model for the Dynamics of Cities" (Barilla, Carlier, Lasry 2021)

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.

Tensorflow implementation of optimal transport (OT) with Sinkhorn algorithm.

FML (Francis' Machine-Learnin' Library) - A collection of utilities for machine learning tasks

Optimal Transport and Optimization related experiments.