ott.solvers.linear.sinkhorn_lr.LRSinkhorn.lse_step