Ai2 · 1 month ago
Senior Research Engineer, OLMo
Seattle, WA
Full-time
Onsite
Mid, Senior Level
$170K/yr - $220K/yr
4+ years expAi2 is a non-profit AI institute dedicated to advancing foundational AI research and innovation. They are seeking a Senior Research Engineer to work on the infrastructure, architecture, modeling, and training of OLMo, with responsibilities including building scalable machine learning pipelines and collaborating on impactful AI research.
Artificial Intelligence (AI)FinanceFinancial ServicesIncubatorsVenture Capital
Responsibilities
Building infrastructure to facilitate the next generation of LLM research
Optimizing training and inference for language models
Triaging between experiments and executing on the most impactful
Supporting and collaborating with an open-source community
Bridging the gap between cutting-edge research and a widely adopted product
Bringing software engineering best practices to a research environment
Releasing your contributions back to the broader community in the form of open source software, model releases, and additions to Ai2’s public API and open research datasets, as well as technical reports
Qualification
Required
4+ years of industry experience building infrastructure that handles data preprocessing/transformation and model training, evaluation, inference, and deployment
Deep experience in the complete model development cycle, including data set construction, training, tuning, evaluation, performance profiling, and monitoring
Knowledge of modern deep learning and natural language processing techniques
Strong software engineering skills, particularly around building performant systems and debugging
Experience with Python and PyTorch/Jax/Tensorflow
Familiarity working with cloud compute resources (e.g. AWS) and containerization (e.g. Docker)
Strong collaboration and communication skills
Preferred
Advanced degree in Data Science/CS/EE/Applied Mathematics/Statistics/ML/NLP or related fields and/or relevant and equivalent engineering experience
Contributions to open-source ML or research libraries (e.g. spaCy, AllenNLP, transformers)
Experience successfully operating models at scale in a production setting
Experience in HPC settings
Curiosity about AI research
Benefits
Team members and their families are covered by medical, dental, vision, basic life insurance, basic accidental death and dismemberment insurance, short-term disability, long-term disability, and an employee assistance program.
Team members are able to enroll in our voluntary life insurance program, our voluntary accidental death and dismemberment program, our health savings account plan, our healthcare reimbursement arrangement plan, and our health care and dependent care flexible spending account plans.
Team members are able to enroll in our company’s 401k plan.
Team members will receive $125 per month to assist with commuting or internet expenses and will also receive $200 per month for fitness and wellbeing expenses.
Team members will also receive up to ten sick days per year, up to seven personal days per year, up to 20 vacation days per year and twelve paid holidays throughout the calendar year.
Team members will be able to receive annual bonuses and can participate in the long-term incentive plan.
Company
Ai2
We are a Seattle-based non-profit AI research institute founded in 2014 by the late Paul Allen.
201-500 employees
H1B Sponsorship
Ai2 has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (20)
2024 (8)
2023 (13)
2022 (8)
2021 (7)
2020 (1)
Funding
Current Stage
Growth StageTotal Funding
$152MKey Investors
NVIDIA
2025-08-14Grant· $152M
Leadership Team
Chris Emura
Executive Director, Head of Applied Science & Technology
David Albright
Head of Design and Communications
Recent News
2025-11-20
Company data provided by crunchbase