Skip to content

ML Research Scientist, Prediction & Smart Agents

Build state-of-the-art ML models to predict traffic behavior for autonomous driving, using generative sequence modeling and controllable agents for planning and simulation. Requires PhD preferred, 2+ years deploying ML systems, and expertise in PyTorch and robotics ML.

194k – 291kMountain View, CAAI ResearchOnsite2+ YOE

About the role

About the Work

  • Design and build scalable, machine learning-based prediction systems to generate multi-modal, realistic, and kinematically feasible trajectories.
  • Conduct cutting-edge research in generative sequence modeling and sequential decision-making, including:
    • Scalable generative sequence modeling approaches.
    • Marginal, conditional, and joint distribution modeling for interactive agents.
    • Transformer-based encoder-decoder architectures.
    • Large generative models and diffusion models.
    • Controllability of agents via conditioning, guidance, and other techniques.
  • Collaborate closely with the Planning team to design realistic and controllable agents for closed-loop simulation, enabling agent training via Reinforcement Learning (RL).
  • Mitigate accumulated uncertainties across interconnected autonomy components.
  • Collaborate across various autonomy teams to develop holistic solutions for top challenges, proposing ideas, prioritizing.
  • Derive practical, deployable solutions and see them deployed on real-world vehicles.

About You

  • Education: M.Sc. or Ph.D. (preferable) in Computer Science, Artificial Intelligence, Mathematics, or related field.
  • Expertise: Research experience in sequential decision-making, prediction, Imitation Learning, Deep Reinforcement Learning, generative modeling, large models, or ML for robotics.
  • Technical Skills: Strong problem solving and programming in Python (required), C++ (beneficial), and ML frameworks like PyTorch.
  • Experience: 2+ years deploying ML systems onboard, ideally in prediction.
  • Publications: Demonstrated research in top conferences (NeurIPS, ICLR, ICML, CVPR, RSS, CoRL, ICRA, IROS).

Nice to have: Deep background in Embodied AI for robotics, Causal reasoning, Model interpretability, Joint prediction and planning, Diffusion Models.

Compensation: Base pay range $193,930 - $291,150, plus annual performance bonus, equity, and benefits.

Skills

PyTorchPythonC++TransformerGenerative ModelsDiffusion ModelsReinforcement LearningImitation LearningSequential Decision-MakingMachine Learning

Similar roles

AI Research jobs

AI Research Resident

AI Research Resident collaborates on research projects developing benchmarks and environments for long-horizon AI agents, identifying model failure modes, and training autonomous agents. Requires current MS/PhD enrollment, RL experience, systems engineering, and strong publications.

200k – 200kSan Francisco, CAAI ResearchRemoteEntry levelBenchmarksFrontier Models

Machine Learning Systems Research Engineer, Agent Post-training - Enterprise GenAI

Develops and optimizes post-training algorithms for agent RL platforms, focusing on LLM training, inference frameworks, and multi-agent systems. Requires 1-3 years production LLM experience, expertise in PyTorch/CUDA, RLHF/PPO, and advanced degree.

218k – 273kSan Francisco, CA +2AI ResearchOn-site1+ YOEPpoCUDA

Machine Learning Research Engineer, Agents - Enterprise GenAI

Develops and deploys state-of-the-art ML models and agents for enterprise GenAI using RL training and post-training algorithms. Requires 1-3 years LLM production experience, RLHF expertise, recent top publications, and advanced CS degree.

218k – 273kSan Francisco, CA +2AI ResearchOn-site1+ YOEPpoLLMs

AI Researcher

Advances AI products through post-training SOTA LLMs using supervised and reinforcement learning techniques on rich query datasets. Owns data pipelines, training frameworks, and model integration while collaborating across teams. Requires 2-6+ years in large-scale LLMs and Python/PyTorch expertise; PhD preferred.

220k – 485kSan Francisco, CA +1AI ResearchOn-site2+ YOESftDpo

Machine Learning Research Scientist: Generative Modeling for Planning

Develops state-of-the-art generative models like diffusion and flow-matching for autonomous planning in self-driving tech. Requires PhD or MSc with 2-3 years experience in generative modeling for robotics, strong Python/C++ skills, and top research publications.

160k – 241kMountain View, CAAI ResearchOn-site2+ YOEC++LLMs