optax.ScaleByRssState# class optax.ScaleByRssState( sum_of_squares: optax.Updates, )[source]# State holding the sum of gradient squares to date.
