optax.ScaleByLionState

Contents

optax.ScaleByLionState#

class optax.ScaleByLionState(count: jax.typing.ArrayLike, mu: optax.Updates)[source]#

State for the Lion algorithm.