Struct MuonWithAuxAdamW 

pub struct MuonWithAuxAdamW { /* private fields */ }

Explicitly routes selected rank-2 parameters to Muon and the exact remainder to AdamW. No shape-based selection occurs.

Implementations

impl MuonWithAuxAdamW

pub fn new( selected: impl IntoIterator<Item = MuonMatrix>, muon_learning_rate: f32, adamw_learning_rate: f32, weight_decay: f32, ) -> Result<Self>

pub fn step(&mut self, model: &mut impl Module) -> Result<()>

pub fn completed_steps(&self) -> i32

pub fn muon(&self) -> &Muon

pub fn adamw(&self) -> &AdamW

Trait Implementations

impl Optimizer for MuonWithAuxAdamW

fn step<M: Module>(&mut self, model: &mut M) -> Result<()>