Neural network trained to solve quantum mechanical problems