Add Transformer-Engine Fused_Adam Optimizer Support#2293

vivekgoe wants to merge 2 commits intopytorch:mainpytorch/torchtitan:mainfrom

vivekgoe:te_fused_adamwvivekgoe/torchtitan:te_fused_adamwCopy head branch name to clipboard