Megatron-DeepSpeed

c20f4d48 - Merge branch 'checkpointing-fix' into 'master'

Merge branch 'checkpointing-fix' into 'master' small fix for CheckpointFunction's backward() method when some args may be NoneType See merge request ADLR/megatron-lm!92