Fix incorrect (1 - b2) coefficient in scale_by_adan#1715

codewithfourtix wants to merge 4 commits into

google-deepmind:maingoogle-deepmind/optax:mainfrom

codewithfourtix:fix-adan-beta2-coefficientcodewithfourtix/optax:fix-adan-beta2-coefficientCopy head branch name to clipboard