megatron.model.language_model.parallel_lm_logits#

megatron.model.language_model.parallel_lm_logits(input_, word_embeddings_weight, parallel_output, bias=None)#

LM logits using word embedding weights.