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

feat(trainer): step CPUOffloadOptimizer in bounded chunks of optimizer state#4154
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.
GitHub Advanced Security / CodeQL succeeded Oct 20, 2026 in 2s

No new alerts in code changed by this pull request

View all branch alerts.