An algorithm to compute Wasserstein barycenter - View it on GitHub