ott.solvers.linear package#

Sinkhorn Solvers#

sinkhorn.solve(geom[, a, b, tau_a, tau_b, rank])

Solve linear regularized OT problem using Sinkhorn iterations.

sinkhorn.Sinkhorn([lse_mode, threshold, ...])

Sinkhorn solver.

sinkhorn.SinkhornOutput([f, g, errors, ...])

Implements the problems.Transport interface, for a Sinkhorn solution.

sinkhorn_lr.LRSinkhorn(rank[, gamma, ...])

A Low-Rank Sinkhorn solver for linear reg-OT problems.

sinkhorn_lr.LRSinkhornOutput(q, r, g, costs, ...)

Implement the problems.Transport interface, for a LR Sinkhorn solution.

Barycenter Solvers#

continuous_barycenter.WassersteinBarycenter([...])

A Continuous Wasserstein barycenter solver, built on generic template.

continuous_barycenter.BarycenterState([...])

Holds the state of the Wasserstein barycenter solver.

discrete_barycenter.discrete_barycenter(geom, a)

Compute discrete barycenter [Janati et al., 2020].

discrete_barycenter.SinkhornBarycenterOutput(f, ...)

Sinkhorn Acceleration#

acceleration.Momentum([start, ...])

Momentum for Sinkhorn updates, either constant [Thibault et al., 2021] or adaptive [Lehmann et al., 2021].

acceleration.AndersonAcceleration([memory, ...])

Implements Anderson acceleration for Sinkhorn.

Implicit Differentiation#

implicit_differentiation.ImplicitDiff([...])

Implicit differentiation of Sinkhorn algorithm.