# ML Research Scientist, Prediction & Smart Agents

**Company:** [Nuro](https://hotfix.jobs/companies/nuro)
**Location:** Mountain View, CA
**Role:** AI Research
**Salary:** $194k – $291k/yr
**Experience:** 2+ years
**Skills:** PyTorch, Python, C++, Transformer, Generative Models, Diffusion Models, Reinforcement Learning, Imitation Learning, Sequential Decision-Making, Machine Learning
**Posted:** 2026-02-19

> 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.

## Job Description

## 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.

## Similar roles

- [AI Research Resident](https://hotfix.jobs/jobs/f50e3e2f-5048-4fcf-b99a-3f98e6bb6efd) - Polymath - Remote - $200k – $200k/yr
- [Research Engineer](https://hotfix.jobs/jobs/9282b9f8-a2cc-4027-9b42-9a8b7d5cf8d1) - Greptile - San Francisco, CA - $180k – $280k/yr
- [AI Research Scientist, New Grad – Agents & Reinforcement Learning](https://hotfix.jobs/jobs/ddd1a923-0761-4cb6-abc6-8fea08d2c270) - Snowflake - Bellevue, WA - $176k – $230k/yr
- [Machine Learning Systems Research Engineer, Agent Post-training - Enterprise GenAI](https://hotfix.jobs/jobs/789d98ee-8cab-42b8-8103-84b486544394) - Scale AI - San Francisco, CA - $218k – $273k/yr
- [Machine Learning Research Engineer, Agents - Enterprise GenAI](https://hotfix.jobs/jobs/86aa8353-26ac-4335-b8ac-c8772791d98e) - Scale AI - San Francisco, CA - $218k – $273k/yr

**Apply:** https://hotfix.jobs/jobs/bfaa61cd-90cb-40f7-a2c7-a1013f29cc69
**Canonical:** https://hotfix.jobs/jobs/bfaa61cd-90cb-40f7-a2c7-a1013f29cc69