SkewAdam cuts MoE optimizer memory by 97.4% and fits 6.78B on a 40GB GPU
Kooky-Ad-4124 · reddit · 2026-07-22
A new preprint introduces SkewAdam, a tiered optimizer designed to slash the memory cost of Mixture-of-Experts training.
- The core idea is to allocate optimizer state by parameter type instead of treating everything equally.
- In the proposed setup, the backbone gets momentum plus factored second moment, experts get factored second moment only, and the router gets exact second moment.
- The paper reports optimizer-state memory falling from 50.6 GB to 1.29 GB, a 97.4% reduction.
- Peak training memory drops from 81.4 GB to 31.3 GB, allowing a 6.78B MoE to fit on a single 40GB GPU without hurting convergence or router stability.
Related event: SkewAdam Slashes MoE Optimizer Memory by 97.4%(2 posts)→
More from Infra
- Analyst: Ant has the most fungible AI workload of any model provider today — BenBajarin · 2026-10-26
- Anthropic S-1: training infra edge cut per-task cost 40% and doubled revenue growth — BenBajarin · 2026-10-26
- Rewriting scaling law study: optimal synthetic data share rises from 10% to 32% with compute — heghbalz · 2026-10-26
- US states halt new data centers as 72% of Americans oppose local builds — YvesMulkers · 2026-10-26
- $5M Seed Round Can't Afford One DeepSeek Training Run: GPU-less AI Startup's Three Paths — FinanceYF5 · 2026-10-26
- HBM demand shifts: low-stack inference plans shelved, Rubin to go 12-high only — BenBajarin · 2026-10-26