Spaces:
Running
Running
File size: 264 Bytes
803ccbf |
1 2 3 4 5 6 7 8 |
# 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.
|