ott.solvers.linear.sinkhorn_lr.LRSinkhorn.kernel_step