DeepSpeed

0ccb2bb6 - Fix gradient is ready with z2 (#7829)

Fix gradient is ready with z2 (#7829) Correctly handle `ds_grad_is_ready` in ZeRO2 --------- Signed-off-by: Olatunji Ruwase <tunji.ruwase@snowflake.com>