code: def plot(runtimes):
      import matplotlib.pyplot as plt
  
      for distributed_setting in runtimes.keys():
          plt.scatter(
              runtimes[distributed_setting].keys(),
              runtimes[distributed_setting].values(),
              label=distributed_setting,
          )
      plt.legend()
      plt.xlabel("Parameters (Billions)")
      plt.ylabel("Training Iteration time (s)")
      plt.title(str("GPT Scaling w/ Offloading"))
      plt.savefig("offload_gpt_scaling.png")
      plt.close()
      if not os.path.exists("/my_workspace/"):
          os.system("mkdir /my_workspace/")
      os.system("cp *.png /my_workspace/")
  
imports: import subprocess
  import os
  from apex.transformer.testing.commons import TEST_SUCCESS_MESSAGE
label_desc: Use image node_id 2 for calling a global function with example usage: plot(runtimes) without return types