optax.ScaleByRmsState

Contents

optax.ScaleByRmsState#

class optax.ScaleByRmsState(
nu: optax.Updates,
)[source]#

State for exponential root mean-squared (RMS)-normalized updates.