code: def _mixed_precision_creator(
      next_creator, shape, dtype, init, context
  ):
      del context
      dtype = _thread_local_state.current_policy.param_dtype
      return next_creator(shape, dtype, init)
  
imports: import collections
  import contextlib
  import threading
  from typing import TypeVar, Optional, Union
  from haiku._src import base
  from haiku._src import data_structures
  from haiku._src import module
  import jmp
label_desc: Use image node_id 8 for calling a global function with example usage: _mixed_precision_creator(next_creator, shape, dtype, init, context) and returns: next_creator