Skip to content

feat(trainer): step CPUOffloadOptimizer in bounded chunks of optimizer state - #4154

Open
rasdani wants to merge 3 commits into
mainfrom
feat/chunked-cpu-offload-optimizer-step
Open

rasdani wants to merge 3 commits into
mainfrom
feat/chunked-cpu-offload-optimizer-step

fix(trainer): size offload step chunks by the largest uneven shard

ab15c01
Select commit
Loading
Failed to load commit list.
Cursor / Cursor Bugbot succeeded Oct 20, 2026 in 4m 8s

Bugbot Review

Bugbot Analysis Progress (4m 13s elapsed)

✅ Gathered PR context (5s)
✅ Completed bug detection (4m 8s)
✅ Posted analysis results (1s)

Final Result: Bugbot completed review - no issues found! ✅

Request ID: 7ec4c3f8-24cf-4e5b-a1af-81f17215b6c8

Details