Skip to content
Ai2Ai2Berkeley, CA

Young Investigator, FlexOlmo

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.

160k – 160k
RemoteAI Research

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

PyTorchLLMsMixture-Of-ExpertsLong-Context Language ModelsRetrievalNatural Language ProcessingMachine LearningScaling LawsData CurationDeep Learning

Similar roles

AI Research jobs
Pindrop

Research Scientist II

PindropUnited States

As a Research Scientist II on the Video team, you will drive core research initiatives, deliver reproducible experimental results, and help translate machine learning models into real-world product solutions, focusing on real-time video processing and deepfake detection.

160k – 185k
Remote3+ YOEAI Research
Together AI

Systems Research Engineer, GPU Programming

Together AISan Francisco, CA

Develops and optimizes GPU-accelerated kernels and algorithms for ML/AI applications, co-designing with modeling, hardware, and software teams. Requires strong GPU programming expertise in CUDA/Triton and knowledge of ML models.

160k – 230k
RemoteAI Research
Nuro

Machine Learning Research Scientist, Behavior Planning and Prediction

NuroMountain View, CA

Develops scalable ML-based planning and prediction systems for autonomous driving trajectories. Requires expertise in sequential decision making, deep RL, imitation learning, generative modeling, and robotics; M.Sc./Ph.D. preferred with top conference publications.

160k – 241k
On-siteAI Research
Coram AI

AI Research Engineer (Robotics)

Coram AISunnyvale, CA

Deploys high-performance vision and multimodal AI models to robotic platforms, optimizing for low-latency inference under hardware constraints. Requires 3+ years in real-time robotics systems, expertise in PyTorch, C++/Rust, and edge deployment.

165k – 210k
On-site3+ YOEAI Research
Ai2

Research Scientist, Robotics

Ai2Seattle, WA

Conducts research on foundation models for robotics, focusing on manipulation, sim-to-real transfer, RL, and skill learning using simulations and real robots. Requires PhD-level expertise in robotics/AI, strong publications, and software skills for open-source breakthroughs.

167k – 261k
On-siteAI Research