# Senior Research Engineer

**Company:** [Ai2](https://hotfix.jobs/companies/ai2)
**Location:** Seattle, WA
**Role:** ML Engineering
**Salary:** $147k – $220k/yr
**Experience:** 4+ years
**Skills:** Python, PyTorch, JAX, TensorFlow, LLMs, Vision-Language Models, Moe, Docker, AWS, GCP
**Posted:** 2026-06-05

> Research Engineer training and scaling flagship open multimodal and agentic models (Olmo, Molmo). Owns end-to-end ML infrastructure, model development, and open-source releases.

## Job Description

## Key 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
- Bringing strong software engineering practices to a research environment and 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)
- Familiarity with cloud and containerization (e.g., GCP, AWS, Docker)
- Strong communication and collaboration skills
- 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)

## Nice-to-Haves
- PhD in ML or equivalent hands-on industry work with deep learning and/or foundation models

## Similar roles

- [Senior Data Scientist - Optimization, Central Market Management & AI](https://hotfix.jobs/jobs/55e6be28-aff3-4f39-b05f-7b2679d31b91) - Lyft - New York, NY - $148k – $185k/yr
- [Senior Data Scientist - Optimization, Central Market Management & AI](https://hotfix.jobs/jobs/a07c6e6d-f87f-4cc2-8465-650dc2e8a4bd) - Lyft - San Francisco, CA - $148k – $185k/yr
- [Senior C++ Software Engineer](https://hotfix.jobs/jobs/2e187e05-ea6e-4e57-acf8-9b3e3d25eeb8) - Applied Intuition - Sunnyvale, CA - $150k – $250k/yr
- [Senior C++ Software Engineer](https://hotfix.jobs/jobs/6fa17829-84c2-459c-af02-99f0a3be9db6) - Applied Intuition - Poway, CA - $150k – $225k/yr
- [Senior C++ Software Engineer](https://hotfix.jobs/jobs/d437d7dc-bb7e-4df2-bf62-7a4391d93214) - Applied Intuition - Ann Arbor, MI - $150k – $200k/yr

**Apply:** https://hotfix.jobs/jobs/2573fe76-af66-4d22-9b1e-ae9ed83e988f
**Canonical:** https://hotfix.jobs/jobs/2573fe76-af66-4d22-9b1e-ae9ed83e988f