Young Investigator, FlexOlmo
Seattle-based nonprofit AI research institute developing openly released models, datasets, tools, and AI applications for science, the environment, and robotics.
Funding history
About Allen Institute for AI (Ai2)
Ai2 is a nonprofit AI research institute founded by Paul Allen in 2014. It develops openly documented language and multimodal models, training data and code, plus applied AI systems for scientific discovery, planetary intelligence, and conservation.
Skills
Candidate Availability
Required and preferred rules are kept separate and reflect the wording in the original posting.
About the Role
You will define and lead a high-impact research project on language models designed for flexible data use, with a focus on mixture-of-experts, long-context models and retrieval. You will train and release models, build open-source software, author scientific papers, and spend about half your time mentoring junior researchers such as PhD students and interns.
Requirements
- PhD or within one year of completing a PhD in Computer Science or similar field
- Research experience in machine learning, NLP, or language and vision
- Outstanding individual contributor skills with deep learning frameworks such as PyTorch
- Strong publication record at AI venues such as NeurIPS, ICLR, ICML, COLM, ACL, EMNLP
- Extensive research experience in large language models, training dynamics, scaling laws and data curation
- Located or willing to relocate to Berkeley, CA
Responsibilities
- Define and lead a high-impact research project
- Train and release leading models
- Build open-source software for the research community
- Author scientific papers for publication
- Mentor junior researchers (50% of time)
- Work as an independent contributor on an Ai2 project (50% of time)
Benefits
- Visa sponsorship available
- Dedicated mentor who is a UC Berkeley faculty member
- Medical, dental, vision, life and AD&D insurance
- Short-term and long-term disability and employee assistance program
- Voluntary life, AD&D, HSA, HRA and FSA plans
- 401k plan
- $125 per month commuting or internet stipend
- $200 per month fitness and wellbeing stipend
- Up to 10 sick days, 7 personal days, 20 vacation days and 12 paid holidays
- Annual bonuses