-
Hi Is there an implementation of Shampoo (Anil et al, 2020) is available on optax? |
Beta Was this translation helpful? Give feedback.
Replies: 2 comments 1 reply
-
Hi! Thanks for the question! The code for this is ready in principle, but we'd like to test it more (for both accuracy and performance) and this is taking slightly longer. I'm currently checking how long this might still take and whether it would make sense to provide it as part of I'll let you know as soon as I know more! |
Beta Was this translation helpful? Give feedback.
-
Hi! Any news about it? The original PR is closed and I didn't see any other PR or discussion about it. |
Beta Was this translation helpful? Give feedback.
Hi!
Thanks for the question! The code for this is ready in principle, but we'd like to test it more (for both accuracy and performance) and this is taking slightly longer. I'm currently checking how long this might still take and whether it would make sense to provide it as part of
experimental
in the meantime.I'll let you know as soon as I know more!