optax.ScaleByRStdDevState

class optax.ScaleByRStdDevState(

mu: optax.Updates,

nu: optax.Updates,

)

State for centered exponential moving average of squares of updates.