Skip to content

Include Lion Optimizer #498

@Haoxiang-Wang

Description

@Haoxiang-Wang

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

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions