Towards Easier-to-Tune Training Algorithms: New Results from AlgoPerf
Organizations: Alice · Goodfire · University of Tübingen · Google
Abstract
Evaluating deep learning optimizer is challenging because confounding factors such as hyperparameter tuning and optimizer-model co-design often obscure true algorithmic performance. We present results of the updated AlgoPerf V1 benchmark, in which we evaluate a new set of training algorithms on wall-clock time-to-target across nine workloads on fixed hardware. This report focuses on the results of the benchmark's self-tuning ruleset, which requires submissions to use a single configuration across all workloads. Out of the 13 submissions in PyTorch and JAX, Schedule-Free AdamW ranked first. In addition to the wall-clock scores, we also analyze the trade-offs between step efficiency and wall-clock execution, revealing that matrix-preconditioned and two-loop methods achieve good sample efficiency but require more aggressive computational optimizations to be competitive in runtime. Finally, our hyperparameter saliency and retention analyses show that the learning rate and momentum are highly sensitive across domains and algorithms, highlighting the need for methods that are more robust against hyperparameter tuning and can generalize well to new tasks.
Figures & tables
| Validation | Base Runtime | |||||
| Task | Dataset | Model | Loss | Metric | Target | Budget (s) |
| Clickthrough rate prediction | Criteo 1TB | DLRMsmall | CE | CE | ||
| MRI reconstruction | fastMRI | U-Net | L1 | SSIM | ||
| Image | ImageNet | ResNet-50 | CE | ER | ||
| classification | ViT | CE | ER | |||
| Speech | LibriSpeech | Conformer | CTC | WER |
| Algorithm | Brief Description | Citation | Submission Code |
| Schedule-Free AdamW v2 (PyTorch) | Couples momentum interpolation with dynamic iterate averaging; eliminates LR schedules and maintains batch norm statistics. This version makes use of the batch-norm fixes in AlgoPerf v1 and is as a result more faithful to the published version of Schedule-Free. | ( Defazio et al., 2024 ) | schedule_free_adamw_v2 |
| AdEMAMix (AdamW equiv.) (PyTorch) | Blends fast and slow gradient EMAs to track short-term fluctuations while retaining long-term directional memory. This submission sets a long warmup for and making it effectively equivalent to AdamW. | ( Pagliardini et al., 2025 ) | ademamix |
| Schedule-Free AdamW (PyTorch) | Legacy Schedule-Free AdamW implementation submitted to the AlgoPerf competition. This version does not contain batch-norm fixes. | ( Defazio et al., 2024 ) | schedule_free_adamw |
| NAdamW (Baseline v0.5) (JAX) | Nesterov-accelerated adaptive gradient optimizer with decoupled weight decay. This version served as the prize qualification baseline for the AlgoPerf v0.5 competition. | ( Dozat, 2016 ; Loshchilov and Hutter, 2019 ) | nadamw_baselinev05 |
| Schedule-Free AdamW v2 (JAX) | JAX implementation of Schedule-Free AdamW v2 with batch-normalization fix. | ( Defazio et al., 2024 ) | schedule_free_adamw_jax_v2 |
| Muon (PyTorch) | Matrix-preconditioned optimizer applying quintic Newton-Schulz iterations on 2D or higher-dimensional non-embedding tensors with AdamW for 1D and embedding parameters. | ( Jordan et al., 2024b ) | muon_torch |
| Tuning Metric | DiLoCo ( ) | NAdamW ( ) | Muon ( ) | AdEMAMix ( ) |
| Golden config score ( ) | 0.786 | 0.989 | 0.927 | 0.583 |
| Oracle score ( ) | 1.279 | 1.509 | 1.973 | 1.403 |
| Retention Rate ( ) | 61.4% | 65.5% | 47.0% | 41.5% |
| Median Speedup ( ) | 0.427 | 0.000 | 0.063 | 0.000 |
| Failure Rate ( ) | 1.0% | 70.0% | 71.0% | 85.0% |
| Framework | Implementation | HPs | Score | |
| JAX | vanilla | JAX | — | |
| PyTorch | vanilla | JAX | ||
| PyTorch | sharded | JAX | ||
| PyTorch | sharded | Torch | ||
| PyTorch | vanilla | Torch |
Appendix figures & tables17 assets
Supplementary material from the paper’s appendix.
Appendix
| Submission | Criteo 1TB | fastMRI | FineWeb-Edu 10B | ResNet | ViT | Conformer | DeepSpeech | OGBG | WMT |
| Schedule-Free AdamW v2 (PyTorch) | 0.56 | 0.27 | — | — | — | 0.85 | 0.76 | 0.37 | 0.43 |
| AdEMAMix (AdamW equiv.) (PyTorch) | 0.59 | 0.37 | 0.38 | — | — | 0.75 | — | 0.37 | 0.72 |
| Schedule-Free AdamW (PyTorch) | 0.77 | 0.29 | — | — | — | — | 0.73 | 0.21 | 0.83 |
| NAdamW (Baseline v0.5) (JAX) | 0.61 | 0.39 | 0.43 | — | — | — | — | 0.24 | 0.62 |
| Schedule-Free AdamW v2 (JAX) | 0.61 | 0.45 | 0.32 | — | 0.88 | — | — | 0.32 | — |
| Muon (PyTorch) | 0.53 | — | 0.32 | — | — | — | — | 0.11 | 0.59 |
| Submission | Criteo 1TB | fastMRI | FineWeb-Edu 10B | ResNet | ViT | Conformer | DeepSpeech | OGBG | WMT |
| JAX submissions | |||||||||
| Schedule-Free AdamW (JAX) | |||||||||
| Schedule-Free AdamW v2 (JAX) | |||||||||
| Muon (JAX) | |||||||||
| NAdamW (JAX) | |||||||||
| NAdamW (Baseline v0.5) (JAX) | |||||||||
| Wall-clock | Steps | ||||
| Submission | Score | Rank | Score | Rank | |
| Schedule-Free AdamW v2 (PyTorch) | 0.5679 | 1 | 0.5253 | 1 | – |
| AdEMAMix (AdamW equiv.) (PyTorch) | 0.5191 | 2 | 0.5146 | 2 | – |
| Schedule-Free AdamW (PyTorch) | 0.4669 | 3 | 0.4461 | 3 | – |
| NAdamW (Baseline v0.5) (JAX) | 0.4506 | 4 | 0.4422 | 4 | – |
| Schedule-Free AdamW v2 (JAX) | 0.4405 | 5 | 0.3945 | 6 | 1 |
| Submission | Criteo 1TB | fastMRI | FineWeb-Edu 10B | ResNet | ViT | Conformer | DeepSpeech | OGBG | WMT |
| Schedule-Free AdamW v2 (PyTorch) | 7,951 | 12,431 | — | — | — | 83,184 | 78,669 | 28,449 | 73,222 |
| AdEMAMix (AdamW equiv.) (PyTorch) | 8,371 | 10,206 | 44,503 | — | — | 67,706 | — | 27,247 | 112,742 |
| Schedule-Free AdamW (PyTorch) | 10,346 | 10,599 | — | — | — | — | 90,588 | 15,853 | 133,517 |
| NAdamW (Baseline v0.5) (JAX) | 9,145 | 6,648 | 47,039 | — | — | — | — | 20,879 | 118,600 |
| Muon (PyTorch) | 7,716 | — | 39,220 | — | — | — | — | 7,711 | 97,224 |
| Schedule-Free AdamW v2 (JAX) | 9,005 | 8,750 | 68,640 | — | 164,206 | — | — | 27,457 | — |
| Standard | 10% relaxed | ||||
| Submission | Score | Rank | Score | Rank | |
| Schedule-Free AdamW v2 (JAX) | 0.4405 | 5 | 0.6852 | 1 | 4 |
| AdEMAMix (PyTorch) | 0.2464 | 12 | 0.6246 | 2 | 10 |
| Schedule-Free AdamW v2 (PyTorch) | 0.5679 | 1 | 0.6072 | 3 | 2 |
| Muon (PyTorch) | 0.4231 | 6 | 0.5993 | 4 | 2 |
| Schedule-Free AdamW (PyTorch) | 0.4669 | 3 | 0.5786 | 5 | 2 |
| Submission | Criteo 1TB | fastMRI | FineWeb-Edu 10B | ResNet | ViT | Conformer | DeepSpeech | OGBG | WMT |
| Schedule-Free AdamW v2 (JAX) | 0.03 | 0.07 | 0.21 | 0.11 | 0.27 | — | — | 0.13 | 0.08 |
| AdEMAMix (PyTorch) | 0.03 | 0.09 | 0.16 | — | — | 0.67 | 0.44 | 0.16 | 0.14 |
| Schedule-Free AdamW v2 (PyTorch) | 0.03 | 0.06 | 0.59 | — | — | 0.61 | 0.42 | 0.16 | 0.06 |
| Muon (PyTorch) | 0.03 | 0.08 | 0.22 | — | — | 0.72 | — | 0.08 | 0.08 |
| Schedule-Free AdamW (PyTorch) | 0.03 | 0.08 | — | — | — | 0.72 | 0.42 | 0.13 | 0.08 |
| Schedule-Free AdamW (JAX) | 0.03 | 0.09 | 0.16 | — | 0.27 | — | — | 0.16 | 0.08 |
| Optimizer | Tuned Hyperparameter | Sampling Range / Values | Distribution | Fixed Parameters & Architectural Notes |
| NAdamW ( ) | Learning rate ( ) | Log-uniform | Cosine decay schedule ( cosine_warmup ) | |
| Momentum | Log-uniform | Fixed , bias correction ( debias = True ) | ||
| Second moment | Log-uniform | Tied dropout rates ( tie_dropouts = True , when supported) | ||
| Weight decay ( ) | Log-uniform | l2_decay_factor = None | ||
| LR warmup fraction ( ) | Discrete uniform | ∗ Tuned only on workloads supporting the regularizer | ||
| Label smoothing ∗ | Discrete uniform |
Deep Learning Optimization
Deep learning optimization develops algorithms and training strategies that help neural networks converge, generalize, and use compute efficiently.
See all 204 papers