code: def init_kv():
      # init kv dns keys
      kv.init(key, mx.nd.ones(shape))
      kv.set_optimizer(mx.optimizer.create("sgd"))
      return kv, kv.rank, kv.num_workers
  
imports: import mxnet
  import json
label_desc: Use image node_id 1 for calling a global function with example usage: init_kv() and returns: kv,  kv,  kv