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