Migrate Reduce on Plateau Learning Rate Scheduler example to the latest flax.nnx module#1711

rajasekharporeddy wants to merge 1 commit into

google-deepmind:maingoogle-deepmind/optax:mainfrom

rajasekharporeddy:reduce_on_plateaurajasekharporeddy/optax:reduce_on_plateauCopy head branch name to clipboard