optax.with_extra_args_support#
- optax.with_extra_args_support(tx: GradientTransformation) GradientTransformationExtraArgs[source]#
Wraps a gradient transformation, so that it ignores extra args.
Wraps a gradient transformation, so that it ignores extra args.