perf(optim): chunked optimizer-state offload for Muon via a subset-step interface (builds on #4427) - #4429

perf(optim): chunked optimizer-state offload for Muon via a subset-step interface (builds on #4427)#4429

faresobeid wants to merge 26 commits into

mainPrimeIntellect-ai/prime-rl:mainfrom

perf/opt-offload-muonPrimeIntellect-ai/prime-rl:perf/opt-offload-muonCopy head branch name to clipboard