Young Investigator, FlexOlmo

Remote Eligibility

Persons in these roles are welcome to work remotely from Berkeley, CA.

$159,650

Ai2 is seeking a talented and motivated Postdoctoral Young Investigator to join the Flex Olmo team, working on a series of large language models designed for flexible data use, with a focus on Mixture‑of‑Experts (MoE), long‑context language models (LCLMs), and retrieval. Postdoctoral Young Investigators will be based in Berkeley, California.

• Define and lead a high‑impact research project.
• Train and release leading models.
• Collaborate with and learn from team members across Ai2.
• Build open‑source software for the research community.
• Author scientific papers for publication in a high‑profile conference or journal.

• Duration: 1‑3 years
• Eligibility: Within one year of completing a Ph.D., or already have a Ph.D.
• The Allen AI Young Investigator is a postdoctoral program offering unique benefits, including mentorship of junior researchers.

• Dedicated Ai2 mentor who is also a faculty member at the University of California, Berkeley.
• 50% work on leading and collaborating on an Ai2 project as an independent contributor.
• 50% work on mentoring junior researchers (Ph.D. students/interns, predoctoral students/interns) and opportunities to receive mentorship in academic activities such as grant writing and teaching, if desired.

• Within one year of completing a Ph.D., or already have a Ph.D., in Computer Science or a related field with research experience in machine learning, natural language processing, language and vision, or related areas.
• Outstanding individual contributor skills, especially with deep learning frameworks (e.g., PyTorch).
• An outstanding publication record at AI‑related venues such as NeurIPS, ICLR, ICML, COLM, ACL, EMNLP; quality of publications in terms of rigor and impact is emphasized.
• Extensive research experience in areas such as large language models, training dynamics, scaling laws, and data curation. Experience with mixture‑of‑experts, long‑context language models, and retrieval is preferred but not required.
• Located in Berkeley, CA, or willing to relocate.

• Must be able to remain in a stationary position for long periods of time.
• Ability to communicate information and ideas clearly to others.
• Ability to observe details at close range.

Ai2 is a Seattle‑based non‑profit AI research institute founded in 2014 by the late Paul Allen. Our mission is to build breakthrough AI to solve the world’s biggest problems. We develop foundational AI research and innovation to deliver real‑world impact through large‑scale open models, data, robotics, conservation, and beyond.

Ai2 is proud to be an Equal Opportunity Employer. We do not discriminate based upon race, religion, color, national origin, sex (including pregnancy, childbirth, or related medical conditions), sexual orientation, gender, gender identity, gender expression, transgender status, age, status as a protected veteran, status as an individual with a disability, or other applicable legally protected characteristics. You may view the related Know Your Rights compliance poster and the Pay Transparency Nondiscrimination Provision by clicking on their corresponding links.

We participate in E‑Verify and will provide the federal government with your Form I‑9 information to confirm that you are authorized to work in the U.S. We are committed to providing reasonable accommodations to employees and applicants with disabilities to the full extent required by the Americans with Disabilities Act (ADA). If you need a reasonable accommodation, please contact us at recruiting.