Add option to disable inverse-pth root statistics summary (doesn't work with static accumulator) and add option to decouple learning rates/weight decay option to match AdaFactor in grafting modes.
Add option to disable inverse-pth root statistics summary (doesn't work with static accumulator) and add option to decouple learning rates/weight decay option to match AdaFactor in grafting modes.

PiperOrigin-RevId: 464208019

google/praxis,google/praxis