Megatron-DeepSpeed

9d264312 - Mlm adaptation (#287)

Mlm adaptation (#287) Co-authored-by: Lintang Sutawika <lintang@datasaur.ai> Co-authored-by: thomasw21 <24695242+thomasw21@users.noreply.github.com> Co-authored-by: Niklas Muennighoff <n.muennighoff@gmail.com>