Learning Multimodal Transition Dynamics for Model-Based Reinforcement Learning