#116 · Primary category: Deep Learning Frameworks
lion-pytorch
🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch
Project last updated:07/09/26
GitHub Stars
2.2K
Forks
55
Contributors
6
License
MIT
Why we included this project
Phil Wang's lion-pytorch is a clean, lightweight port of the Lion optimizer that Google Brain discovered through evolutionary search. You can drop it into an existing PyTorch training loop with a single import and test whether its sign-momentum update actually beats your current AdamW baseline. The README carries the practical tuning guidance from the paper, most importantly a learning rate that should run 3 to 10 times lower than AdamW's, paired with a correspondingly larger decoupled weight decay. It also collects honest field reports from researchers who tried it, including the finding that holding the learning rate constant made Lion underperform Adam, while dividing it by roughly ten produced better early results. That mix of a faithful reference implementation and real-world caveats helps practitioners decide whether the optimizer is worth adopting in their own projects.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
tensorflow
An Open Source Machine Learning Framework for Everyone
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
keras
Deep Learning for humans
nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
ray
Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.