https://medium.com/@monocosmo77/working-with-transport-maps-part5-machine-learning-optimal-transport-107124db6da1