Before revision:  We then train a Mixture-of-Experts within each cluster to predict initial guesses and provide a comparison with modality-agnostic learning.
After revision:  We then train a Mixture-of-Experts within each cluster to predict state and control trajectories to warm-start the optimal control solver and provide a comparison with modality-agnostic learning.