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
- llama.cpp lands Flash Attention tuning for RDNA4, big prefill gains on AMD — pmttyji · 2026-09-11
- Your p99 latency benchmark may be lying: a deep dive into coordinated omission — Franc0Fernand0 · 2026-09-11
- Running MiniMax H3 on 12GB VRAM: quantization, Turbo LoRAs and attention backends compared — Possible_Mood676 · 2026-09-11
- Spomin: live KV cache compaction squeezes 500k tokens of context into 180k resident — wgaca2 · 2026-09-11
- PiPNN nearest-neighbor search wins three awards, up to 78x faster index building — khademinori · 2026-09-11
- M.2-Oculink eGPU Link Silently Downgrades to PCIe Gen1 — Here's How to Check — El_90 · 2026-09-11