optax.ScaleByRmsState

Contents

optax.ScaleByRmsState#

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

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