src/transformers/models/wav2vec2/modeling_flax_wav2vec2.py/FlaxConvWithWeightNorm/_get_normed_weights
class FlaxConvWithWeightNorm:    def _get_normed_weights(self):
        weight_v_norm = jnp.linalg.norm(self.weight_v, axis=(0, 1))[None, None, :]
        normed_weight_v = jnp.divide(self.weight_v, weight_v_norm)
        normed_kernel = jnp.multiply(normed_weight_v, self.weight_g)
        return normed_kernel