Add BF16 stochastic rounding option for optimizers#1124

gau-nernst merged 7 commits intopytorch:mainpytorch/ao:mainfrom

gau-nernst:bf16_optim_srgau-nernst/ao:bf16_optim_srCopy head branch name to clipboard