Machine learning for the solution of the Schrödinger equation