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