MESH: Memory-Efficient Sinkhorn Optimization for Mixture-of-Experts Training
MoE训练新方案:Sinkhorn梯度下降替代AdamW,减少优化器状态内存,让大模型训练更省显存。
arXiv:2608.04407v1 Announce Type: cross Abstract: Memory-efficient matrix optimizers such as Sinkhorn gradient descent remove most AdamW optimizer sta…