DeepSpeed

8d9c9e74 - Scale query and key before attn_score gemm for more accurate attention