ott.solvers.quadratic.gw_barycenter.GromovWassersteinBarycenter.update_state