optax.schedules.InjectHyperparamsState

class optax.schedules.InjectHyperparamsState(

count: jnp.ndarray,

hyperparams: dict[str, jax.typing.ArrayLike],

inner_state: base.OptState,

)

Deprecated class kept for backwards compatibility.

Deprecated since version 0.1.9: Use InjectStatefulHyperparamsState instead.