def METHOD_NAME(self, logits, **kwargs):
    """
    Project the logits with shape ```[batch_size, vocab_size]``` into
    label_word_logits with shape ```[batch_size, num_label_words]```.
    """
    raise NotImplementedError