code: def simple_dtype(dtype) -> str:
      if isinstance(dtype, type):
          dtype = dtype(0).dtype
      dtype = dtype.name
      dtype = dtype.replace("complex", "c")
      dtype = dtype.replace("double", "d")
      dtype = dtype.replace("float", "f")
      dtype = dtype.replace("uint", "u")
      dtype = dtype.replace("int", "s")
      return dtype
  
file_path: dm-haiku/haiku/_src/utils.py
imports: import collections.abc
  from collections.abc import Sequence
  import decimal
  import inspect
  import pprint
  import re
  from typing import Any, TypeVar, Union
  import jax
label_desc: Use image node_id 10 for calling a global function with example usage: simple_dtype(dtype) and returns: dtype