Break up grpo_train to be more functional/composable to enable R&D
Then implementing e.g. REINFORCE becomes a matter of swapping out loss function, rollout config
Also lets us easily experiment with ideas like expert demonstrations, curriculum learning, etc. These Composable primitives we can assemble into GRPO, REINFORCE, PPO, …