[Feat]: Add batch_update and replicate_ns_matrix_axes to Muon optimizer - #1
Draft
chiaotung97 wants to merge 1 commit into
Draft
[Feat]: Add batch_update and replicate_ns_matrix_axes to Muon optimizer#1chiaotung97 wants to merge 1 commit into
chiaotung97 wants to merge 1 commit into