Research Engineer, FlexOlmo

AI2
Seattle
Workplace: OnsiteFull time$ 128,880 - 193,320 annuallyFunction: Research & Scientific (R&D)Experience: 4+ yearsEducation: bachelorsSkills: ["Collaboration","Communication","Problem-solving","Teamwork"]

Research Engineer focused on building and scaling infrastructure for next-generation foundation models, with emphasis on Mixture-of-Experts and long-context language models. You’ll own design and implementation of training systems, optimize training/inference, bridge research to product, and contribute open-source software and model releases from a Seattle-based team.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
AI2
AI2
3 months ago

Research Engineer, FlexOlmo

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 22 hours agoStatus: Live

Job Summary

Research Engineer focused on building and scaling infrastructure for next-generation foundation models, with emphasis on Mixture-of-Experts and long-context language models. You’ll own design and implementation of training systems, optimize training/inference, bridge research to product, and contribute open-source software and model releases from a Seattle-based team.
Location: Seattle
Workplace: Onsite
Employment Type: Full time
Job Function: Research & Scientific (R&D)

Key Responsibilities

  • •Build infrastructure to facilitate the next generation of LLM research.
  • •Optimize training and inference for language models.
  • •Triaging between experiments and executing on the most impactful work.
  • •Bridge the gap between cutting-edge research and a widely adopted product.
  • •Bring software engineering best practices to a research environment.

Pay and Benefits

Salary: $ 128,880 - 193,320 annually
Perks:MedicalDentalVision401kLife InsuranceDisabilityPaid LeavePersonal DaysCommuting StipendGym Membership

Key Requirements

  • •Bachelor's degree in Data Science/CS/EE/Applied Mathematics/Statistics/ML/NLP, or equivalent relevant experience, and expertise at building ML infrastructure.
  • •4+ years of experience building infrastructure that handles data preprocessing/transformation and model training, evaluation, inference, and deployment; experience across the complete model development cycle.
  • •Knowledge of modern deep learning and natural language processing techniques.
  • •Strong software engineering skills, particularly around building performant systems and debugging.
  • •Must have experience with Python and PyTorch/Jax/TensorFlow and be comfortable picking up new languages/libraries; familiarity with AWS and Docker.
Experience:4+ yearsAINLPOpen SourceResearch
Education:Bachelor's
Skills:CollaborationCommunicationProblem-solvingTeamwork
Languages:En
Tech Stack:PythonPyTorchJaxTensorFlowAWSDocker

Company Brief

AI2
The Allen Institute for AI (AI2) is a nonprofit research institute that builds AI systems and conducts fundamental research to contribute to scientific discovery, natural language understanding, and open-source tools for the AI community.
Industry: AI & Machine Learning
Company Size: Medium (51 to 250 employees)
Growth: Nonprofit & NGO
Funding: Grant Funded
Headquarters: Seattle, United States
Founded: 2014
WebsiteLinkedIn