[torchtrain] add gradient scaler#25

tianyu-l merged 3 commits intogh/tianyu-l/1/basepytorch/torchtitan:gh/tianyu-l/1/basefrom

gh/tianyu-l/1/headpytorch/torchtitan:gh/tianyu-l/1/headCopy head branch name to clipboard