Adding mask for weight decay in Prodigy-AdamW implementation#1702

copybara-service[bot] merged 1 commit into

maingoogle-deepmind/optax:mainfrom

test_931966597google-deepmind/optax:test_931966597Copy head branch name to clipboard