Skip to content

Young Investigator, FlexOlmo

160k – 160kBerkeley, CAAI ResearchRemote
Summary

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.

About the role

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
Skills
PyTorchLarge Language ModelsMixture-of-ExpertsLong-Context Language ModelsRetrievalNatural Language ProcessingMachine LearningScaling LawsData CurationDeep Learning
Similar roles at this salary range
All AI Research jobs →
Snowflake

AI Research Scientist, New Grad – Agents & Reinforcement Learning

Conduct research on autonomous AI agents and reinforcement learning to build self-improving systems that reason, code, and learn at scale within the Snowflake Data Cloud. Requires a PhD (or equivalent) and strong expertise in RL and agentic AI.

176k – 230kBellevue, WAAI ResearchOn-siteEntry levelJAXDPO
Snowflake

Post-Doctoral Researcher

Post-doctoral researcher conducting independent and collaborative AI/ML research focused on high-impact domains like medicine, finance, and law. Requires a recent or imminent PhD and publications in top venues.

160k – 220kBellevue, WAAI ResearchHybridEntry levelJAXRAG
SpotOn

Senior Software Engineer - Python/Typescript

Senior engineer building AI-driven automation systems to replace manual business workflows across operations, sales, and support. Requires 7+ years experience, production Python/TypeScript skills, and 1-2 years building agentic AI systems.

160k – 190kChicago, IL +3AI ResearchHybrid7+ YOEAWSLLMs
Datology AI

Research Engineer

As a Research Engineer, you will conduct and enable cutting-edge research, translating it into the core product pipeline. You will develop and improve state-of-the-art data curation strategies, accelerating research and ensuring product innovation.

180k – 300kRedwood City, CAAI ResearchOn-site4+ YOEML ModelsAI Models
Reltio

Sr. Staff Enterprise AI Architect

As a senior architectural leader within the Enterprise AI Hub, you will design and implement scalable AI systems aligned to Reltio’s enterprise AI strategy. You will partner with the Director, Enterprise AI to evolve the foundational architecture that enables secure, governed agentic workflows across the organization.

132k – 244kUnited StatesAI ResearchRemote12+ YOERBACABAC