Young Investigator, FlexOlmo

Ai2 is seeking a talented and motivated Postdoctoral Young Investigator to join the FlexOlmo team, working on large language models designed for flexible data use, with a focus on Mixture-of-Experts (MoE), long-context language models (LCLMs), and retrieval.

The Young Investigator position is a postdoctoral program offering unique benefits, enabling you to balance working collaboratively on an Ai2 project while having opportunities to mentor junior researchers.

Responsibilities:

• 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

Requirements:

• Within one year of completing PhD or already have a PhD in Computer Science or similar field with research experience in machine learning, natural language processing, language and vision, or related areas
• Outstanding individual contributor (IC) skills, especially with deep learning frameworks (e.g., PyTorch)
• Outstanding publication record at AI-related venues
• Extensive research experience in areas such as large language models, training dynamics, scaling laws, and data curation
• Located in or willing to relocate to Berkeley, CA

Benefits:

• Medical, dental, vision, basic life insurance, basic accidental death and dismemberment insurance, short-term disability, long-term disability, and employee assistance program
• Voluntary life insurance program, voluntary accidental death and dismemberment program, health savings account plan, healthcare reimbursement arrangement plan, and health care and dependent care flexible spending account plans
• 401k plan
• $125 per month for commuting or internet expenses
• $200 per month for fitness and wellbeing expenses
• Up to ten sick days per year, up to seven personal days per year, up to 20 vacation days per year, and twelve paid holidays throughout the calendar year
• Annual bonuses