Full-time position
Senior Research Engineer, Olmo + Molmo
Allen Institute for Artificial Intelligence (Ai2)
- Seattle, United States · On-site
- Artificial Intelligence, Machine Learning and
About
Ai2, the non-profit Allen Institute for AI, trains fully open foundation models such as Olmo, Molmo, Tulu and FlexOlmo. This research engineer is a core member of the team training Ai2's flagship open models, owning delivery from system design to release: building and optimising infrastructure for large language model, multimodal and agentic research (training and inference pipelines, dataset curation, large-scale preprocessing); designing, training and evaluating vision-language models and agentic workflows with tool use, planning and long-horizon tasks; scoping and leading research projects; and supporting the open-source community through model releases, datasets, public APIs and technical reports.
The role is based in Seattle with a base salary range of $174,240-$261,360.
Requirements
Eight years of industry experience including four or more years of ML infrastructure (data preprocessing, training, evaluation, inference, deployment); end-to-end model development experience; familiarity with LLMs (MoEs, long-context) and vision-language models; knowledge of agentic systems; strong software engineering in Python with PyTorch, JAX or TensorFlow; cloud and container experience. BS or MSc in computer science, statistics, engineering, applied mathematics or a related field, or equivalent experience.
Topics
Source: Allen Institute for Artificial Intelligence (Ai2)