def METHOD_NAME(
    scope,
    x,
    num_groups=32,
    group_size=None,
    epsilon=1e-6,
    dtype=jnp.float32,
    bias=True,
    scale=True,
    bias_init=initializers.zeros_init(),
    scale_init=initializers.ones_init(),