code: def test_gpt_neox_zero1() -> None:
      config = conf.load_config(
          conf.deepspeed_examples_path("gpt_neox/zero1.yaml")
      )
      config = conf.set_max_length(config, {"batches": 100})
      config = conf.set_min_validation_period(config, {"batches": 100})
      # Changing to satisfy cluter size and gpu mem limitations.
      config = conf.set_slots_per_trial(config, 8)
      config["hyperparameters"]["conf_file"] = [
          "350M.yml",
          "determined_cluster.yml",
      ]
      config["hyperparameters"]["overwrite_values"][
          "train_batch_size"
      ] = 32
  
      exp.run_basic_test_with_temp_config(
          config, conf.deepspeed_examples_path("gpt_neox"), 1
      )
  
file_path: determined/e2e_tests/tests/nightly/test_distributed.py
imports: import os
  import shutil
  import tempfile
  import warnings
  import pytest
  from determined.common import util
  from tests import config
  from tests import experiment
label_desc: Use image node_id 4 for calling a global function with example usage: test_gpt_neox_zero1() without return types