def METHOD_NAME(self):
    setup_serial_env()
    net = DanglingExt()
    args = SimpleNamespace(local_rank=0)
    engine, _, _, _ = 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()