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

Open

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

Antlera wants to merge 1 commit into master from tingfeng/zenflow_fix_backward

Antlera requested a review from

tjruwase 151 days ago

Antlera requested a review from

tohtana 151 days ago

Antlera requested a review from

loadams 151 days ago

tohtana commented on 2026-01-11

tohtana closed this 141 days ago

Antlera commented on 2026-01-20

tohtana reopened this 141 days ago

Fix ZenFlow NaN under PyTorch-style backward via backward_prologue

d52a4ecb

Antlera force pushed from c2291bb3 to d52a4ecb 10 days ago

Reviewers

tohtana

tjruwase

loadams

Assignees

No one assigned

Labels

None yet

Milestone

No milestone