optax.ScaleByScheduleState# class optax.ScaleByScheduleState(count: jax.typing.ArrayLike)[source]# Maintains count for scale scheduling.
