Hello @kellerjordan0 @_arohan_. I noticed that in your recent optimizer work, you appear to have used the inefficient versions of Muon and Shampoo that have long since been succeeded by PowerWash last week. The new algorithm is quite simple and elegant: it merely generates a set of weights with a different seed and evaluates until one of them passes the validation threshold, therefore cutting speedrun time down to 0 steps. The SplittingHairs normalization addition is particularly useful for stabilizing performance. I hope we can collaborate to bring this new standard into broader usage!
Neural MMO creator Joseph Suarez parodies recent ML optimizer research, joking that a new "PowerWash" method achieves zero-step training
The fictional algorithm evaluates random weights until validation passes.
Users praise the PowerWash zero-step algorithm satire for its clear explanations of optimizer techniques along with clever humorous code details like fused weight updates.
Most Activity
I love power wash, cosigned
Hello @kellerjordan0 @_arohan_. I noticed that in your recent optimizer work, you appear to have used the inefficient versions of Muon and Shampoo that have long since been succeeded by PowerWash last week. The new algorithm is quite simple and elegant: it merely generates a set of weights with a different seed and evaluates until one of them passes the validation threshold, therefore cutting speedrun time down to 0 steps. The SplittingHairs normalization addition is particularly useful for stabilizing performance. I hope we can collaborate to bring this new standard into broader usage!

Now that I have your attention, any suggestions on our ~200 line CUDA implementation of Muon would be greatly appreciated https://github.com/PufferAI/PufferLib/blob/4.0/src/muon.cu. In the 5.0 branch on the same file, I played with a small change to preserve LR across model sizes, but there have not been any major improvements otherwise.

@jsuarez i love the fused weight update and fp32 upcasts. excellent code, sir

@jsuarez @kellerjordan0 @_arohan_ real alpha is in taking all of those hf checkpoints and learning to initialize, maybe also a better optimization basis

@jsuarez @kellerjordan0 @_arohan_ thank you for explaining this so clearly. I feel like explanations like this are remarkably rare, and I look at a lot of explanations.

@jsuarez @kellerjordan0 @_arohan_ serious?

@jsuarez @kellerjordan0 @_arohan_ what is this PowerWash? can you link it?

@jsuarez @kellerjordan0 @_arohan_ Bruh

@_arohan_ power washing is one of those things that scratches a brain itch nobody knew they had until they tried it

@jsuarez 😂😂

@michalwols @jsuarez @kellerjordan0 @_arohan_ 🤔

@jsuarez @kellerjordan0 @_arohan_ lol