Fix clip_by_global_norm producing NaN when the global norm and max_norm are both zero - #1752

Fix clip_by_global_norm producing NaN when the global norm and max_norm are both zero#1752

shashvat-singham wants to merge 1 commit into

google-deepmind:maingoogle-deepmind/optax:mainfrom

shashvat-singham:fix/clip-by-global-norm-zero-nanshashvat-singham/optax:fix/clip-by-global-norm-zero-nanCopy head branch name to clipboard