Megatron-DeepSpeed

5467a59e - don't do `compare('max_position_embeddings')` for alibi