Megatron-DeepSpeed

c3be5d3f - Combine Specs (#304)

Combine Specs (#304) * Add support for weighted train * Combine attn_mask dropping & data fromat specs Co-authored-by: thomasw21 <24695242+thomasw21@users.noreply.github.com>