Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
-
Updated
Jul 16, 2024 - Python
Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
FML (Francis' Machine-Learnin' Library) - A collection of utilities for machine learning tasks
Tensorflow implementation of optimal transport (OT) with Sinkhorn algorithm.
Code for ECML/PKDD paper: "LSMI-Sinkhorn: Semi-supervised Mutual Information Estimation with Optimal Transport"
Implementation of the paper "Anticipation-Free Training for Simultaneous Machine Translation"
Code for the paper Geodesic Sinkhorn for Fast and Accurate Optimal Transport on Manifolds.
[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.
Implementation of DETR in JAX
Add a description, image, and links to the sinkhorn topic page so that developers can more easily learn about it.
To associate your repository with the sinkhorn topic, visit your repo's landing page and select "manage topics."