contrib: add SOAP optimizer (arXiv:2409.11321)#1692

irhyl wants to merge 3 commits into

google-deepmind:maingoogle-deepmind/optax:mainfrom

irhyl:contrib/soap-optimizerirhyl/optax:contrib/soap-optimizerCopy head branch name to clipboard