# Notes

Files copied from [google-research/scalable_shampoo/optax](https://github.com/google-research/google-research/tree/master/scalable_shampoo/optax).

Imports have been modified to be relative.

This will be replaced with `optax-shampoo` package eventually.