https://medium.com/@monocosmo77/working-with-optimal-transport-maps-part1-advanced-computation-cf93f90bd27e