Senior Research Engineer, Olmo + Molmo
The Allen Institute for Artificial Intelligence2 months ago
Base Salary
$174k - $261k/yr
Responsibilities
- Build and optimize infrastructure for LLM, multimodal, and agentic research, including training and inference pipelines, dataset curation, and large-scale preprocessing.
- Design, train, and evaluate multimodal vision-language models and agentic workflows involving tool use, planning, and long-horizon tasks.
- Scope and lead research projects and prioritize high-impact experiments.
- Apply strong software engineering practices to research and bridge cutting-edge work to production-quality products.
- Contribute to the open-source community through model releases, datasets, public APIs, and technical reports.
Requirements
- At least 8 years of industry experience, including 4 or more years of ML infrastructure experience covering preprocessing, training, evaluation, inference, and deployment.
- End-to-end model development experience, including dataset construction, training, fine-tuning, evaluation, profiling, and monitoring.
- Experience with LLM architectures such as mixture-of-experts and long-context models, as well as vision-language models such as Molmo and LLaVA.
- Knowledge of agentic systems, including tools, memory, and long-running workflows.
- Strong software engineering fundamentals and experience building performant, scalable systems and debugging confidently.
- Proficiency in Python and a major ML framework such as PyTorch, JAX, or TensorFlow.
- Familiarity with cloud and containerization technologies such as GCP, AWS, and Docker.
- BS or MSc in Computer Science, Statistics, Engineering, Applied Mathematics, or a related quantitative field, or equivalent experience.
- At least 2 years of software development experience or equivalent experience.
- Strong communication and collaboration skills.
Benefits
- Medical, dental, vision, employee assistance, health savings account, healthcare reimbursement, and flexible spending account plans.
- 401(k) plan.
- $125 per month for commuting or internet expenses and $200 per month for fitness and wellbeing expenses.
- Up to 10 sick days, 7 personal days, 20 vacation days, and 12 paid holidays annually.
- Annual bonuses and participation in a long-term incentive plan.
- Seattle office-based role with on-site requirements varying by position and team.
- Learning opportunities including Ai2 Academy lectures, guest speakers, and ongoing education support.
- Generous paid vacation, sick leave, and family leave, with an emphasis on work-life balance.
Tech Stack
Categories
AI ResearchML Engineering
About The Allen Institute for Artificial Intelligence
The Allen Institute for Artificial Intelligence (AI2) is a Seattle-based nonprofit research institute that advances AI through open models, datasets, and tools, and runs applied projects in NLP, vision, robotics, and conservation. It creates widely used resources such as Semantic Scholar and AllenNLP, and operates the AI2 Incubator to spin out startups. Founded in 2014 by Paul Allen, it partners with academia and industry and publishes open research.