https://medium.com/@monocosmo77/working-with-zeldovich-approximation-part4-machine-learning-2024-db733d593989