Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add Adan optimizer. #1090

Merged
merged 1 commit into from
Oct 14, 2024
Merged

Conversation

carlosgmartin
Copy link
Contributor

@carlosgmartin carlosgmartin commented Oct 4, 2024

Picking up #410.

optax/_src/alias.py Outdated Show resolved Hide resolved
optax/_src/transform.py Outdated Show resolved Hide resolved
optax/_src/alias.py Outdated Show resolved Hide resolved
@fabianp
Copy link
Member

fabianp commented Oct 4, 2024

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

@carlosgmartin carlosgmartin force-pushed the adan branch 2 times, most recently from 537bc1f to e7480c4 Compare October 4, 2024 22:25
@carlosgmartin
Copy link
Contributor Author

@fabianp Done.

@fabianp
Copy link
Member

fabianp commented Oct 7, 2024

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!

@carlosgmartin
Copy link
Contributor Author

@fabianp

image

Copy link
Member

@fabianp fabianp left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

perfect, thanks!!!

@copybara-service copybara-service bot merged commit 4e0a5c7 into google-deepmind:main Oct 14, 2024
8 checks passed
@carlosgmartin carlosgmartin deleted the adan branch October 14, 2024 18:31
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants