def METHOD_NAME(cls, args, dictionary, embed_dim, path=None):
    def _vocab_init(tensor, **kwargs):
        nn.init.normal_(tensor, mean=0, std=embed_dim**-0.5)
        nn.init.constant_(tensor[1], 0)
    embed_tokens = VocabParallelEmbedding(
        len(dictionary), embed_dim, dictionary.pad(), init_method=_vocab_init
    )
    return embed_tokens