sinkhorn 开源项目推荐（54 个）

fwilliams /point-cloud-utils

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

ott-jax /ott

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

ot-triton-lab /flash-sinkhorn

The official repository of FlashSinkhorn [ICML 2026 Oral]

fwilliams /fml

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

bassrehab /mhc-visualizer

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

gabsens /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.

MarkMoHR /TF-OT-Sinkhorn

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

csyanbin /LSMI-Sinkhorn

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

vienmai /drot

Douglas-Rachford Splitting for Optimal Transport

GiulsLu /OT-gradients

该项目在 GitHub 上没有填写描述

KrishnaswamyLab /GeoSinkhorn

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

dmivilensky /Wasserstein-distance-calculation

Optimal Transport and Optimization related experiments.

MasterSkepticista /detr

JAX implementation of DETR

asigalov61 /Music-Transformers-Library

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

George0828Zhang /sinkhorn-simultrans

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

chansigit /torchgw

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

LawrenceMMStewart /Semi-Supervised-Learning-with-OT

该项目在 GitHub 上没有填写描述

hrshtv /CTreeOT

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

svdrecbd /mhc-mlx

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

ManifoldFR /entropic-mfg

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