src/transformers/models/t5/convert_t5x_checkpoint_to_pytorch.py/t5x_layer_norm_lookup
def t5x_layer_norm_lookup(params, i, prefix, layer_name):
    """Returns the layer norm param of a layer."""
    return params[f"{prefix}/layers_{i}/{layer_name}/scale"]