code: def load_backend(
      model,
      dataset,
      tokenizer,
      cfg_train,
      cfg_impl,
      elapsed_time=0.0,
      setup=_default_setup,
  ):
      if cfg_impl.name == "torch-default":
          return initialize_torch(
              model,
              dataset,
              tokenizer,
              cfg_train,
              cfg_impl,
              elapsed_time,
              setup=setup,
          )
      elif cfg_impl.name == "deepspeed":
          return initialize_deepspeed(
              model,
              dataset,
              tokenizer,
              cfg_train,
              cfg_impl,
              setup=setup,
          )  # cannot resume
      else:
          raise ValueError(f"Invalid backend {cfg_impl.name} given.")
  
imports: import torch
  from .torch_default import initialize_torch
  from .deepspeed_integration import initialize_deepspeed
label_desc: Use image node_id 1 for calling a global function with example usage: load_backend(model, dataset, tokenizer, cfg_train, cfg_impl, elapsed_time, setup) and returns: initialize_torch, initialize_deepspeed