examples/research_projects/longform-qa/eli5_utils.py/RetrievalQAEmbedder/embed_questions
class RetrievalQAEmbedder:    def embed_questions(self, q_ids, q_mask, checkpoint_batch_size=-1):
        q_reps = self.embed_sentences_checkpointed(q_ids, q_mask, checkpoint_batch_size)
        return self.project_q(q_reps)