https://medium.com/@monocosmo77/using-gross-pitaevskii-equation-in-machine-learning-part1-a82c95a3bb7f