code: def is_jax_device_array(inst):
      """
      Check whether `inst` is instance of jax device array
      """
      if inst_has_typename(inst, ["jaxlib", "xla_extension", "Array"]):
          return True
      if inst_has_typename(
          inst, ["jaxlib", "xla_extension", "DeviceArray"]
      ):
          return True
      return False
  
file_path: aim/aim/sdk/num_utils.py
label_desc: Use image node_id 6 for calling a global function with example usage: is_jax_device_array(inst) and returns: False, True, True