[2/2] Add grouped-expert FP8 GraphTrainer support - #2

[2/2] Add grouped-expert FP8 GraphTrainer support#2

WhoisZihan wants to merge 2 commits into

feature/graph-trainer-fp8-support-pr1WhoisZihan/torchtitan:feature/graph-trainer-fp8-support-pr1from

feature/graph-trainer-fp8-support-pr2WhoisZihan/torchtitan:feature/graph-trainer-fp8-support-pr2Copy head branch name to clipboard