iamwyldecat's picture
refactor(muon): change argument adam_wd to weight_decay and handle params' type
02ac540