ott.solvers.linear package
Contents
ott.solvers.linear package#
Sinkhorn Solvers#
|
Solve linear regularized OT problem using Sinkhorn iterations. |
|
Sinkhorn solver. |
|
Implements the problems.Transport interface, for a Sinkhorn solution. |
|
A Low-Rank Sinkhorn solver for linear reg-OT problems. |
|
Implement the problems.Transport interface, for a LR Sinkhorn solution. |
Barycenter Solvers#
A Continuous Wasserstein barycenter solver, built on generic template. |
|
Holds the state of the Wasserstein barycenter solver. |
|
Compute discrete barycenter [Janati et al., 2020]. |
|
Sinkhorn Acceleration#
|
Momentum for Sinkhorn updates, either constant [Thibault et al., 2021] or adaptive [Lehmann et al., 2021]. |
|
Implements Anderson acceleration for Sinkhorn. |
Implicit Differentiation#
Implicit differentiation of Sinkhorn algorithm. |