optax.ScaleByRssState

Contents

optax.ScaleByRssState#

class optax.ScaleByRssState(sum_of_squares: optax.Updates)[source]#

State holding the sum of gradient squares to date.