https://medium.com/@monocosmo77/revisiting-continuous-normalizing-flows-part10-molecular-machine-learning-2024-a0f1815a0ffe