code: def init_kv():
      # init kv dns keys
      kv.init(keys, [mx.nd.ones(shape)] * len(keys))
      kv.init("9", mx.nd.ones(shape))
      kv.init("10", mx.nd.ones(shape))
      kv.init("99", mx.nd.ones(big_shape))
      kv.init("100", mx.nd.ones(big_shape))
      # worker info
      my_rank = kv.rank
      nworker = kv.num_workers
      # init updater on servers
      kv.set_optimizer(mx.optimizer.create("test", rescale_grad=rate))
      return kv, my_rank, nworker
  
file_path: mxnet/tests/nightly/dist_device_sync_kvstore.py
imports: import sys
  import mxnet
  import numpy
  import numpy.random
  import time
label_desc: Use image node_id 2 for calling a global function with example usage: init_kv() and returns: kv,  my_rank,  nworker