DeepSpeed

Fix no-grad grad-fn lookup in ZeRO hook counting on PyTorch 2.3 (#7830)

#7841

Merged