Jen Wei mengalami spiral kematian learning rate saat menguji AdamW, Muon, dan implementasi terbaru Dion3 di dalam OLMo-core untuk sebuah talk Konferensi PyTorch mendatang.
- Proses debugging mengungkapkan masalah yang melibatkan state optimizer terdistribusi dan penjadwalan LR.
- Sebuah jebukan aliasing tensor PyTorch tertentu diidentifikasi sebagai penyebab utama ketidakstabilan.
- Wei mendokumentasikan seluruh petualangan debugging tersebut di Medium untuk membantu orang lain menghindari jebukan memori serupa di optimizer terdistribusi.