https://medium.com/@monocosmo77/working-with-transport-maps-part7-machine-learning-optimal-transport-2200fce07e8e