ott.experimental.mmsinkhorn.MMSinkhornOutput.cost_fns