optax.ScaleByLionState

class optax.ScaleByLionState(

count: jax.typing.ArrayLike,

mu: optax.Updates,

)

State for the Lion algorithm.