Geometrization of the Schrödinger equation