This is your work, valued
LLM Pretrain
flash-muon. Flash-Muon: An Efficient Implementation of Muon Optimizer