code: def init_to_zero(name):
      return lambda nargs: nargs[name].zero_()
  
file_path: apex/apex/contrib/openfold_triton/_mha_kernel.py
imports: import triton
  import triton.language
label_desc: Use image node_id 1 for calling a global function with example usage: init_to_zero(name) and returns: function