-
|
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.
-
|
I'm not in any big hurry but I did want to try it out for one of my problems. It'll be great to have it as an Thanks! |
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
experimentalin the meantime.I'll let you know as soon as I know more!