code: def test_ext_param_return():
      setup_serial_env()
  
      net = DanglingExt()
  
      args = SimpleNamespace(local_rank=0)
      engine, optim, _, _ = deepspeed.initialize(
          args=args,
          model=net,
          model_parameters=net.parameters(),
          config=config,
      )
  
      for _ in range(5):
          input = torch.rand(net.dim).to(engine.device).half()
          loss = engine(input)
          engine.backward(loss)
          engine.step()
  
imports: import os
  from types import SimpleNamespace
  import torch
  import pytest
  import deepspeed
  from deepspeed.runtime.zero.partition_parameters import ZeroParamStatus, partitioned_param_data_shape
  import deepspeed.comm
  from unit.common import DistributedTest, get_master_port
label_desc: Use image node_id 5 for calling a global function with example usage: test_ext_param_return() without return types