fix: shard single-row gate weights along columns for Muon - #3723

Bugbot Analysis Progress (1m 25s elapsed)

✅ Gathered PR context (6s)
✅ Completed bug detection (1m 19s)
✅ Posted analysis results (1s)

Final Result: Bugbot completed review - no issues found! ✅

Request ID: serverGenReqId_2af8e19c-9a7e-4da5-b13a-042b58f6f382