A look back at optimizer search: Lion boosted ViT accuracy 2% and cut JFT pretraining compute 5x

akbirthko · x · 2026-10-24

The author revisits research on discovering optimizers via program search, citing Lion (Symbolic Discovery of Optimization Algorithms, arXiv:2302.06675) and VeLO.

Lion, from Google, keeps only momentum and is more memory-efficient than Adam: up to 2% ImageNet accuracy gains for ViT, up to 5x pretraining compute savings on JFT, and up to 2.3x less compute for diffusion training. The author notes that varying costs and value estimates would be an interesting follow-up problem.

Original post →

More from Research

Research channel →