Saurav Maheshkar

> Two extra small changes: > > * remove the entity = "saurav" and default to None (so it' uses your default entity after calling `wandb.login`) > * Add the...

> extra detail, use a relative path to the image, so when we merge the path to your branch is not needed The image is not in my fork of...

CI is green again 😄

Gentle ping: @fabianp

> Hi @SauravMaheshkar , > > a couple more comments after discussing this issue with the dev team: > > * let's promote this to the main package, since normalized...

@fabianp I've refactored my branch to add the transform to main rather than contrib. Request for Review 😄

I've added a doctest 👍🏽 @fabianp

> looks good to me, but I forgot to ask you one (hopefully) last thing: > > please add this new function to the API documentation: https://github.com/google-deepmind/optax/blob/main/docs/api/transformations.rst > > thanks!...

> duplicated #1000 since I'm having some issues getting this merged due to internal errors Typically copybara I assume 😅

> yup. Finally merged 🎉 Thanks a ton. Feels good to have it merged. Now on to #652