optax.ScaleBySM3State

Contents