optax.TransformUpdateExtraArgsFn# class optax.TransformUpdateExtraArgsFn( *args, **kwargs, )[source]# An update function accepting additional keyword arguments.
