muon_exps
CUDA kernels for the Muon optimizer: Newton-Schulz polar iteration (quintic + Gram-NS variants), benchmarked and verified against torch.opti
Star 0
Fork 0
Watch
Open Closed
New
No issues yet.