Implement SOAP optimizer (#1580)#1617

jinukuntlaakhilakumargoud-web wants to merge 2 commits intogoogle-deepmind:maingoogle-deepmind/optax:mainfrom

jinukuntlaakhilakumargoud-web:implement-soap-1580jinukuntlaakhilakumargoud-web/optax:implement-soap-1580Copy head branch name to clipboard