ott.solvers.linear#
Linear solvers are the bread-and-butter of OT solvers. They can be called on
their own, either the Sinkhorn
ott.solvers.linear.sinkhorn.Sinkhorn
or Low-Rank
ott.solvers.linear.sinkhorn_lr.LRSinkhorn
solvers, to match two
datasets. They also appear as subroutines for more advanced solvers in the
ott.solvers
module, notably ott.solvers.quadratic
or
ott.solvers.nn
.
Sinkhorn Solvers#
|
Solve linear regularized OT problem using Sinkhorn iterations. |
|
Sinkhorn solver. |
|
Holds the state variables used to solve OT with Sinkhorn. |
|
Implements the problems.Transport interface, for a Sinkhorn solution. |
|
A Low-Rank Sinkhorn solver for linear reg-OT problems. |
|
State of the Low Rank Sinkhorn algorithm. |
|
Implement the problems.Transport interface, for a LR Sinkhorn solution. |
Barycenter Solvers#
Continuous Wasserstein barycenter solver [Cuturi and Doucet, 2014]. |
|
Holds the state of the Wasserstein barycenter solver. |
|
A Wasserstein barycenter solver for histograms on a common geometry. |
|
Sinkhorn Acceleration#
|
Momentum for Sinkhorn updates. |
|
Implements Anderson acceleration for Sinkhorn. |
Implicit Differentiation#
Implicit differentiation of Sinkhorn algorithm. |