Unofficial implementation of https://arxiv.org/abs/2412.13148 which appears to be inspired by https://github.com/KellerJordan/Muon and related work by https://x.com/jxbz
this is also labeled with annotations to show how the two are similar/different