code: def _argmax(x, axis, keepdims):
      # FIXME find_domain
      return torch.argmax(x, axis, keepdim=keepdims)
  
imports: import numbers
  import typing
  import torch
  import funsor.ops
label_desc: Use image node_id 12 for calling a global function with example usage: _argmax(x, axis, keepdims) and returns: torch