code: def ds_implementation(vals, gamma, epsilon):
      global inference_module
      if inference_module is None:
          inference_module = InferenceBuilder().load()
      return inference_module.rms_norm(vals, gamma, epsilon)
  
file_path: DeepSpeed/tests/unit/ops/transformer/inference/test_rms_norm.py
imports: import deepspeed
  import torch
  import pytest
  from deepspeed.accelerator import get_accelerator
  from deepspeed.ops.op_builder import InferenceBuilder
  from .inference_test_utils import allclose, get_dtypes
label_desc: Use image node_id 2 for calling a global function with example usage: ds_implementation(vals, gamma, epsilon) and returns: inference_module