Research Engineer, FlexOlmo

The Allen Institute for Artificial Intelligence is seeking a Research Engineer to work on next-generation large language model architectures, focusing on Mixture-of-Experts and long-context models. This role involves designing and implementing scalable machine learning pipelines for foundational model research and contributing to open-source AI development.

• Design and implement next-generation LLM architectures
• Build scalable machine learning pipelines
• Optimize training and inference for language models
• Collaborate with researchers and contribute to open-source AI
• Focus on Mixture-of-Experts and long-context language models
• Requires 4+ years of ML infrastructure experience with Python and AWS