code: def make_hk_transform_ignore_jax_transforms(f):
      """Wraps f such that if it was jit/pmap-ed Haiku won't realise."""
      return lambda *a, **k: f(*a, **k)
  
imports: from collections.abc import Mapping, Sequence
  import dataclasses
  import functools
  import pprint
  from typing import Any, Callable, Optional, TypeVar, Union
  from haiku._src import base
  from haiku._src import data_structures
  from haiku._src import module
  from haiku._src import transform
  from haiku._src import utils
  import jax
  import jax.numpy
  import numpy
  import tabulate
label_desc: Use image node_id 4 for calling a global function with example usage: make_hk_transform_ignore_jax_transforms(f) and returns: function