boris's picture
feat: support pod (#139)
803ccbf unverified
|
raw
history blame
264 Bytes

Notes

Files copied from google-research/scalable_shampoo/optax.

Imports have been modified to be relative.

This will be replaced with optax-shampoo package eventually.