search

google-deepmind / optax

Optax is a gradient processing and optimization library for JAX.
https://optax.readthedocs.io
Apache License 2.0
1.68k stars 190 forks source link

Implementation of Eve Optimizer #805

Open amosyou opened 8 months ago

amosyou commented 8 months ago

Adds the Eve optimizer in wrapper form in response to #475

fabianp commented 8 months ago

@amosyou thanks for working on this! I see this is in draft mode, let me know when you'd like one of us to talk a look into it!