boris's picture
feat: update shampoo
9ecdd3f

A newer version of the Streamlit SDK is available: 1.44.1

Upgrade

Notes

Files copied from google-research/scalable_shampoo/optax.

Imports have been modified to be relative.

This will eventually be replaced with optax-shampoo package.