Model-based reinforcement learning with neural network dynamics