DeepSpeed

Avoid graph breaks in torch.compile caused by inner classes in the backward hooks

#7062

Merged