# Young Investigator, FlexOlmo

**Company:** [Ai2](https://hotfix.jobs/companies/ai2)
**Location:** Remote
**Role:** AI Research
**Salary:** $160k – $160k/yr
**Skills:** PyTorch, LLMs, Mixture-Of-Experts, Long-Context Language Models, Retrieval, Natural Language Processing, Machine Learning, Scaling Laws, Data Curation, Deep Learning
**Posted:** 2026-04-22

> Postdoctoral researcher leading high-impact AI projects on Mixture-of-Experts and long-context language models, training/releasing models, building open-source tools, publishing papers, and mentoring juniors. Requires recent PhD in CS/ML with strong publication record and PyTorch expertise.

## Job Description

## 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 high-profile conferences or journals.
- 50% work leading and collaborating on Ai2 project as an independent contributor.
- 50% work mentoring junior researchers (PhD students/interns, predoctoral students/interns).

## Requirements
- Within one year of completing PhD or already have PhD in Computer Science or similar field.
- Research experience in machine learning, natural language processing, language and vision, or related areas.
- Outstanding individual contributor skills, especially with deep learning frameworks (e.g. **PyTorch**).
- Outstanding publication record at AI-related venues (NeurIPS, ICLR, ICML, COLM, ACL, EMNLP).
- Extensive research experience in large language models, training dynamics, scaling laws, and data curation.
- Experience with mixture-of-experts, long-context language models, and retrieval preferred.
- Located or willing to relocate to Berkeley, CA.

## Compensation
- $159,650

## Similar roles

- [Research Scientist II](https://hotfix.jobs/jobs/b15dadf9-a6d4-4c68-abd5-cc569b90c2f3) - Pindrop - Remote - $160k – $185k/yr
- [Systems Research Engineer, GPU Programming](https://hotfix.jobs/jobs/7ee52a5b-008e-4ea8-a30f-cf95d53d5c23) - Together AI - Remote - $160k – $230k/yr
- [Machine Learning Research Scientist, Behavior Planning and Prediction](https://hotfix.jobs/jobs/27236e08-fa94-4679-b1a8-9f47d582fa06) - Nuro - Mountain View, CA - $160k – $241k/yr
- [AI Research Engineer (Robotics)](https://hotfix.jobs/jobs/ae0a1d4c-01fe-45ee-b3c9-f82ab3101f91) - Coram AI - Sunnyvale, CA - $165k – $210k/yr
- [Research Scientist, Robotics](https://hotfix.jobs/jobs/bc1e0e8e-1934-418b-bed7-f2cf89a4379c) - Ai2 - Seattle, WA - $167k – $261k/yr

**Apply:** https://hotfix.jobs/jobs/e0f4781d-9e05-48dd-8784-cba76809e17f
**Canonical:** https://hotfix.jobs/jobs/e0f4781d-9e05-48dd-8784-cba76809e17f