2d Optimal Transport Python, .