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