Senior Research Engineer, Olmo + Molmo
Ai2 · Seattle, WA · 1 mo ago
Engineering$147k/yrFull-time
About the role
Persons in these roles are expected to work from our offices in Seattle. On-site requirements vary based on position and team.
Responsibilities
- Building and optimizing infrastructure for LLM, multimodal, and agentic research — including training/inference pipelines, dataset curation, and large-scale preprocessing
- Designing, training, and evaluating multimodal models (vision + language) and agentic workflows, including tool use, planning, and long-horizon tasks
- Scoping and leading research projects, prioritizing experiments for highest impact
- Bridging cutting-edge work to production-quality products
- Contributing to and supporting the open-source community through model releases, datasets, public APIs, and technical reports
Requirements
- 4+ years of ML infrastructure experience — data preprocessing, model training, evaluation, inference, and deployment
- Experience with end-to-end model development — dataset construction, training, fine-tuning, evaluation, profiling, and monitoring
- Familiarity with modern model architectures — including LLMs (MoEs, long-context models), vision-language models (e.g., Molmo, LLaVA), and experience training and evaluating both
- Agentic systems knowledge — tools, memory, and long-running workflows
- Strong software engineering fundamentals — performant, scalable systems and confident debugging
- Proficiency in Python and a major ML framework (PyTorch, JAX, or TensorFlow), with the flexibility to pick up new tools as needed
- Familiarity with cloud and containerization (e.g., GCP, AWS, Docker)
- Strong communication and collaboration skills — we're a small, close-knit team and work best when everyone's pulling in the same direction
Qualifications
- BS or MSc in Computer Science, Statistics, Engineering, Applied Mathematics, or a related quantitative field (or equivalent experience)
- A minimum of 2 years of software development experience. (or equivalent experience)
Skills
- Deep learning and/or foundation models expertise
- Software engineering fundamentals
- Cloud and containerization familiarity
- Communication and collaboration skills
Benefits
- Medical, dental, vision coverage
- Health savings account plan, healthcare reimbursement arrangement plan, and health care and dependent care flexible spending account plans
- 401k plan
- Commuting and internet expense allowance, fitness and wellbeing expense allowance
- Sick days, personal days, vacation days, paid holidays
- Annual bonuses and participation in the long-term incentive plan
Pay
The base salary range is $146,880 - $220,320, and in addition we have generous bonus plans to provide a competitive compensation package.
Schedule
On-site requirements vary based on position and team.