code: def multi_gpu(model, gpus=None, cpu_merge=True, cpu_relocation=False):
      """Takes as input the model, and returns a model
      based on the number of GPUs available on the machine
      or alternatively the 'gpus' user input.
  
      NOTE: this needs to be used before model.compile() in the
      model inputted to Scan in the form:
  
      from talos.utils.gpu_utils import multi_gpu
      model = multi_gpu(model)
  
      """
  
      from tensorflow.keras.utils import multi_gpu_model
  
      return multi_gpu_model(
          model,
          gpus=gpus,
          cpu_merge=cpu_merge,
          cpu_relocation=cpu_relocation,
      )
  
label_desc: Use image node_id 2 for calling a global function with example usage: multi_gpu(model, gpus, cpu_merge, cpu_relocation) and returns: multi_gpu_model