POT Python Optimal Transport
École Polytechnique · Centre de Mathématiques Appliquées de l'École polytechnique · +29 more institutions
Abstract
This new release contains several new features and bug fixes. Among the new features we have a new submodule ot.batch that contains batch parallel solvers for several OT problems including Sinkhorn, Gromov-Wasserstein and Fused Gromov-Wasserstein. This new submodule can be used to solve many independent OT problems in parallel on CPU or GPU with shared source and target support sizes. We also implemented a new Nystrom kernel approximation for the Sinkhorn solver that can be used to speed up the computation of the Sinkhorn divergence on large datasets. We also added new 1D solvers for Linear circular OT and new solvers for free support OT barycenters with generic cost functions and for barycenters between…
Citation impact
- FWCI
- —
- Percentile
- —
- References
- 0
Authors
21- RFRémi FlamaryCorresponding
École Polytechnique, Centre de Mathématiques Appliquées de l'École polytechnique
- VCVincent-Cuaz, Cédric
Institut de Recherche en Informatique et Systèmes Aléatoires
- NCNicolas Courty
Université de Bretagne Sud
- AGAlexandre Gramfort
Institut national de recherche en sciences et technologies du numérique, Laboratoire d'Informatique, du Traitement de l'Information et des Systèmes
- KOKachaiev, Oleksii
University of Genoa
Topics & keywords
- Python (programming language)
- Computer science
- Programming language