Megatron-DeepSpeed

7181de45 - Add a test that the packed_masking works in the modeling side