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

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

FlashSinkhorn: IO-Aware Entropic Optimal Transport in PyTorch + Triton. Streaming Sinkhorn with O(nd) memory.

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

Interactive visualization of Manifold-Constrained Hyper-Connections (mHC) for stable deep network training

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.

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

Douglas-Rachford Splitting for Optimal Transport

Optimal Transport and Optimization related experiments.

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

JAX implementation of DETR

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

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

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

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

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

MLX + Metal implementation of mHC: Manifold-Constrained Hyper-Connections by DeepSeek-AI.