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.

The official repository of FlashSinkhorn [ICML 2026 Oral]

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.

JAX implementation of DETR

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

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

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

TorchGW — Fast Sampled Gromov-Wasserstein optimal transport in pure PyTorch. GPU-accelerated with Triton fused Sinkhorn kernels. 3-175x faster than POT.

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

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

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