Learning Compositional Koopman Operators for Model-Based Control