Optimizer comparison

Contour visualization inspired by Alec Radford — SGD, Adam, AdamW and Muon

Step 0
Click on the surface to set a custom start point
3

Learning rates