examples/research_projects/longform-qa/eli5_utils.py/RetrievalQAEmbedder/embed_answers
class RetrievalQAEmbedder:    def embed_answers(self, a_ids, a_mask, checkpoint_batch_size=-1):
        a_reps = self.embed_sentences_checkpointed(a_ids, a_mask, checkpoint_batch_size)
        return self.project_a(a_reps)