ott.solvers.linear#
Linear solvers are the bread-and-butter of OT solvers. They can be called on
their own, either the Sinkhorn
Sinkhorn
or Low-Rank
LRSinkhorn
solvers, to match two
datasets. They also appear as subroutines for more advanced solvers in the
ott.solvers
module, notably ott.solvers.quadratic
.
Sinkhorn Solvers#
|
Solve linear regularized OT problem using Sinkhorn iterations. |
|
Sinkhorn solver. |
|
Holds the state variables used to solve OT with Sinkhorn. |
|
Holds the output of a Sinkhorn solver applied to a problem. |
|
Low-Rank Sinkhorn solver for linear reg-OT problems. |
|
State of the Low Rank Sinkhorn algorithm. |
|
Transport interface for a low-rank 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. |
|
Other Solvers#
|
Univariate solver to compute 1D OT distance over slices of data. |
Sinkhorn Acceleration#
|
Momentum for Sinkhorn updates. |
|
Implements Anderson acceleration for Sinkhorn. |
Implicit Differentiation#
Implicit differentiation of Sinkhorn algorithm. |
|
Wrapper around JAX native linear solvers. |
|
|
Wrapper around lineax solvers. |
Low-rank Sinkhorn Utilities#
|
Dykstra's algorithm for the unbalanced |
|
Dykstra's algorithm for the unbalanced |