Remote Postdoc Investigator: MoE & LLM Research Lead

The Allen Institute for Artificial Intelligence; Berkeley (CA); Remote; USD 136,000 - 184,000; Full time

Benefits offered by this job

Visa sponsorship

Job summary

The Allen Institute for Artificial Intelligence (Ai2) invites applications for a Postdoctoral Young Investigator to join the FlexOlmo team in Berkeley, California. Work on MoE, long-context models, and retrieval while building open-source tools and publishing impactful research.

Duration is 1–3 years with opportunities to mentor researchers and collaborate across Ai2. Salary is $159,650 with visa sponsorship available for eligible candidates.

Qualifications

• PhD within one year of completion or already earned in Computer Science or related field with ML/NLP experience.
• Outstanding IC skills with deep learning frameworks (e.g. PyTorch) and a strong publication record.
• Experience with large language models, MoE, retrieval is preferred but not required.

Responsibilities

• Define and lead a high-impact research project.
• Train and release leading models.
• Collaborate with and learn from team members across Ai2.
• Build open-source software for the research community.
• Author scientific papers for publication in a high-profile conference or journal.

Skills

Deep learning (PyTorch)

Education

PhD

Tools

PyTorch

Job description

The Allen Institute for Artificial Intelligence (Ai2) invites applications for a Postdoctoral Young Investigator to join the FlexOlmo team in Berkeley, California. Work on MoE, long-context models, and retrieval while building open-source tools and publishing impactful research.

Duration is 1–3 years with opportunities to mentor researchers and collaborate across Ai2. Salary is $159,650 with visa sponsorship available for eligible candidates.