← all repositories

lucidrains/lion-pytorch

A PyTorch implementation of Lion, a neural network optimizer discovered via evolutionary search by Google Brain that claims improvements over AdamW for training deep learning models.

2.2k stars Python ML Frameworks
lion-pytorch
Velocity · 7d
+1.8
★ / day
Trend
steady
star history

This repository provides a PyTorch implementation of the Lion optimizer (Evolved Sign Momentum), an optimization algorithm for training neural networks that was automatically discovered through genetic program search by Google Brain. Lion claims to outperform AdamW on various benchmarks while being simpler in form. The implementation includes configurable hyperparameters for learning rate, weight decay, and beta values, along with guidance on adapting training schedules from AdamW to Lion.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.