https://medium.com/@mathcube7/gradient-descent-for-the-schr%C3%B6dinger-equation-with-python-90524af3d00b