boris's picture
feat: support pod (#139)
803ccbf unverified
|
raw
history blame
264 Bytes
# 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.