🛠 Tech Stack

💼 About This Role

You'll lead a high-impact research project on large language models with FlexOlmo team at Ai2. You'll train and release leading models while authoring scientific papers.

🎯 What You'll Do

• Define and lead a high-impact research project
• Train and release leading models
• Collaborate with team members across Ai2
• Build open-source software for the research community
• Author scientific papers for publication

📋 Requirements

• PhD in Computer Science or similar field within 1 year
• Outstanding individual contributor skills with deep learning frameworks like PyTorch
• Outstanding publication record at AI venues like NeurIPS, ICLR
• Research experience in large language models, training dynamics

✨ Nice to Have

• Experience with mixture-of-experts
• Experience with long-context language models
• Experience with retrieval

🎁 Benefits & Perks

• 🏖️ Flexible start date
• 📚 Opportunity to mentor junior researchers
• 🎓 Collaboration with UC Berkeley and BAIR lab
• 🛂 Visa sponsorship available
• 🧑‍🏫 Dedicated mentor who is UC Berkeley faculty