Learning Compositional Dynamics Models For Model-Based Control