SOAP & SPlus scale pretraining across 96+ DiLoCo workers via update whitening

aaron_defazio · x · 2026-10-20

Aiming at labs that can't buy enough co-located chips from a single provider, new work shows SOAP and SPlus optimizers achieve strong worker scaling to 96+ DiLoCo workers by whitening the update's covariance. This suggests frontier-scale pretraining can run across heterogeneous, non-co-located compute from multiple providers. Details in a 9-part thread.

Related event: SOAP/SPlus whitening enables stable pretraining across 96 DiLoCo nodes(2 posts)→

Original post →

More from Infra

Infra channel →