optax icon indicating copy to clipboard operation
optax copied to clipboard

Published 20 hours ago • verified publisher icon google-deepmind

Add Adan optimizer.

Open carlosgmartin opened this issue 1 year ago • 4 comments

Picking up #410.

carlosgmartin avatar Oct 04 '24 01:10 carlosgmartin

thanks for the contribution @carlosgmartin ! Left some minor documentation nitpicks in there

fabianp avatar Oct 04 '24 07:10 fabianp

@fabianp Done.

carlosgmartin avatar Oct 04 '24 22:10 carlosgmartin

looks good to me. just one last thing. could you please paste here a screenshot (or two, if it doesn't fit) of the rendering of the adan optimizer's docstring in the webpage? thanks!

fabianp avatar Oct 07 '24 07:10 fabianp

@fabianp

image

carlosgmartin avatar Oct 07 '24 16:10 carlosgmartin