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.