-
Notifications
You must be signed in to change notification settings - Fork 310
Open
Description
Google Brain releases a new optimizer, Lion (https://arxiv.org/abs/2302.06675) which shows superior performance than AdamW on a wide range of deep learning training tasks. They provide a PyTorch implementation at https://github.com/google/automl/tree/master/lion
@lucidrains also implemented a version at https://github.com/lucidrains/lion-pytorch
Could you include it in this codebase? @jettify
Metadata
Metadata
Assignees
Labels
No labels