https://medium.com/@monocosmo77/working-with-schr%C3%B6dinger-bridge-part3-machine-learning-2024-1778e5a92ee3