code: def transformation(
      gen, fun: WrappedFun, *gen_static_args
  ) -> WrappedFun:
      """Adds one more transformation to a WrappedFun.
  
      Args:
        gen: the transformation generator function
        fun: a WrappedFun on which to apply the transformation
        gen_static_args: static args for the generator function
      """
      return fun.wrap(gen, gen_static_args, None)
  
imports: from __future__ import annotations
  from functools import partial
  import operator
  from typing import Any, Callable, NamedTuple
  import weakref
  from jax._src import config
  from jax._src import core
  from jax._src import traceback_util
  from jax._src.tree_util import tree_map
  from jax._src.util import curry
label_desc: Use image node_id 1 for calling a global function with example usage: transformation(gen, fun) and returns: fun