DeepSpeed

7ed952ef - Fix bug in bfloat16 optimizer related to checkpointing (#4434)

Fix bug in bfloat16 optimizer related to checkpointing (#4434) * fix: bf16 optimizer if condition * fix: unexpected keyword argument 'load_serial' * fix: add load_serial arg to bf16_optimizer * style: fix indentation --------- Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>