optax.schedules.InjectHyperparamsState

optax.schedules.InjectHyperparamsState#

class optax.schedules.InjectHyperparamsState(count: jnp.ndarray, hyperparams: dict[str, jax.typing.ArrayLike], inner_state: base.OptState)[source]#

Deprecated class kept for backwards compatibility.

Deprecated since version 0.1.9: Use InjectStatefulHyperparamsState instead.