fix the outdated end2end training examples of moe+torchtitan#3242

rakkit wants to merge 1 commit intopytorch:mainpytorch/ao:mainfrom

rakkit:fix_moe_training_examplesrakkit/ao:fix_moe_training_examplesCopy head branch name to clipboard