code: def wrap_init(f, params=None) -> WrappedFun:
      """Wraps function `f` as a `WrappedFun`, suitable for transformation."""
      params = () if params is None else tuple(sorted(params.items()))
      return WrappedFun(f, (), (), params, None, 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 4 for calling a global function with example usage: wrap_init(f, params) and returns: WrappedFun