https://medium.com/@monocosmo77/working-with-transport-maps-part6-machine-learning-optimal-transport-e6f4326cfff1