DeepSpeed

bf16_optimizer: fixes to different grad acc dtype

#6485

Merged