Skip to content

Extend muon to support mask_callable weight dimension numbers#1643

Open
dylantirandaz wants to merge 1 commit into
google-deepmind:mainfrom
dylantirandaz:muon-mask-callable-support
Open

Extend muon to support mask_callable weight dimension numbers#1643
dylantirandaz wants to merge 1 commit into
google-deepmind:mainfrom
dylantirandaz:muon-mask-callable-support

Commits

Commits on Mar 23, 2026