DeepSpeed

Replace torch.jit.script with torch.compile (#7835)

#7840

Merged