DeepSpeed
[AutoTP] Enable HF `colwise_gather_output` to support `lm_head` replace
#8146
Merged
Merged