Commit History
-
fix(muon): handle un-distributed env
1f13dae -
refactor(muon): change argument adam_wd to weight_decay and handle params' type
02ac540
-
fix(muon): free tensors that are no longer needed
64757cb -
chore(muon): update comment
036642a -
chore(muon): clean build and update doc
febdf5b
-
fix(muon): delete intermediate tensors immediately to lower peak mem usage
bdd2678 -
chore: initial commit
8535e80