optax.scale_by_optimistic_gradient

optax.scale_by_optimistic_gradient(alpha: jax.typing.ArrayLike = 1.0, beta: jax.typing.ArrayLike = 1.0) → optax.GradientTransformation

Compute generalized optimistic gradients.

See optax.optimistic_adam_v2(), optax.optimistic_gradient_descent() for more details.

Parameters:

• alpha – Coefficient for generalized optimistic gradient descent.

• beta – Coefficient for negative momentum.

Returns:

A optax.GradientTransformation object.