code: def t5x_layer_norm_lookup(params, i, prefix, layer_name):
      """Returns the layer norm param of a layer."""
      return params[f"{prefix}/{prefix}/{layer_name}/scale"][:, i]
  
file_path: transformers/src/transformers/models/umt5/convert_umt5_checkpoint_to_pytorch.py
imports: import argparse
  import collections
  import numpy
  import torch
  from flax import traverse_util
  from t5x import checkpoints
  from transformers import MT5Config, UMT5EncoderModel, UMT5ForConditionalGeneration
  from transformers.utils import logging
label_desc: Use image node_id 4 for calling a global function with example usage: t5x_layer_norm_lookup(params, i, prefix, layer_name) and returns: params