partition 1 partition 2 partition 3 partition 4 optimizer step teacher loaded, waiting for rollouts