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