Fix: ZenFlow Adam integration for updated PyTorch backward flow (#7759)#7771

Open

Antlera wants to merge 8 commits intomasterfrom

tingfeng/zenflow_fix_backward

Commits on Jan 12, 2026

Refine logic and error handling based on feedback
Antlera
committed
Merge branch 'tingfeng/zenflow_fix_backward' of github.com:deepspeedai/DeepSpeed into tingfeng/zenflow_fix_backward
Antlera
committed

Commits on Jan 20, 2026

Fix loss scaling logic in DeepSpeedEngine.
Antlera
committed

Commits on Jan 21, 2026

Merge branch 'master' into tingfeng/zenflow_fix_backward
Antlera
authored