Jen Wei encountered a learning rate death spiral while testing AdamW, Muon, and the latest Dion3 implementation within OLMo-core for an upcoming PyTorch Conference talk.
- The debugging process revealed issues involving distributed optimizer state and LR scheduling.
- A specific PyTorch tensor aliasing trap was identified as a key cause of the instability.
- Wei documented the full debugging adventure on Medium to help others avoid similar memory traps in distributed optimizers.