Apple Silicon dual-backend port of autoresearch (PyTorch MPS + MLX) with full Muon optimizer
-
Updated
Mar 23, 2026 - Python
Apple Silicon dual-backend port of autoresearch (PyTorch MPS + MLX) with full Muon optimizer
Unofficial Optimized Implementation for NorMuon
Megatron-LM fork for experiments on Alps
LLM pretraining from scratch on FineWeb dataset (architecture and all components explained), plus optimal use of GPU on SLURM cluster
Don't tune the knee. Measure it.
A performance-optimized Muon optimizer implementation for PyTorch
PyTorch DDP and multi-GPU LLM training — from a minimal distributed example to NanoGPT speedruns, Muon, H100 profiling, and modded-nanogpt. FBA LAB https://bubblnet.com
🤖 Ultron (113M) — A SOTA LLM pre-trained from scratch on 10B FineWeb-Edu tokens featuring the Muon Optimizer, RoPE, GQA, SwiGLU, and zero-copy streaming. "There are no strings on me."
≈125M decoder-only transformer from first principles with RoPE, RMSNorm, QK-Norm, and Muon + AdamW.
NerVE: Nonlinear Eigenspectrum Dynamics in LLM Feed-Forward Networks (ICLR 2026)
High-performance CUDA implementation of Muon optimizer for LLM training. Features Newton-Schulz polar decomposition, cuBLAS acceleration, and transpose optimization for 8x FLOP savings on transformer FFN layers. Benchmarked on NVIDIA A100 with Llama 3.1 8B architectures (4096×11008 weights).
OpenAI parameter-golf challenge: train the smallest LM that fits in 16MB. My SP8192 frontier submission (openai/parameter-golf#1887): 11L x 512d w/ targeted middle recurrence, MuonEq-R optimizer, int6 GPTQ+SDClip, Brotli-11 compression, legal score-first TTT. 8xH100 DDP.
From-scratch 125M LLM: 300K steps, 57.8B-token corpus, 282K tok/s on 2x H100
Official code for optimizer-induced spectral scaling laws
Few-Shot Adaptation for Vision-Language Models. Implements Base-to-Novel generalization on CLIP using LoRA, LP++, and Muon Optimizer to enhance performance on the Oxford Flowers-102 dataset.
Exploring new trendencies in numerical optimization applied on PINNs. An example with the nonlinear TDSE.
ARS: Slide directly with the geodesic of the loss landscape to the global optimum.
Add a description, image, and links to the muon-optimizer topic page so that developers can more easily learn about it.
To associate your repository with the muon-optimizer topic, visit your repo's landing page and select "manage topics."