Research Scientist, Simulation Agents

About the role

The Behaviors team at Waabi builds cutting‑edge simulation agents and scenario‑generation algorithms for Waabi World, the company’s simulation platform. As a Research Scientist you will create realistic, controllable agents that push the self‑driving system to its limits and form a core part of the scientific safety case.

Key responsibilities

• Own and pursue a research agenda to develop realistic and controllable simulation agents.
• Advance the state‑of‑the‑art in imitation learning, reinforcement learning, generative models, foundation models, planning and search for simulation agents.
• Collaborate with simulation, autonomy, and safety teams to define high‑impact research problems, ship solutions into production, and drive progress toward Waabi’s milestones.
• Mentor junior scientists and interns, fostering a culture of scientific rigor and rapid experimentation.
• Publish high‑impact research at top‑tier conferences in machine learning or robotics.

Required profile

• Masters or PhD in machine learning, computer science, engineering, or a related field.
• Strong background in imitation learning and/or reinforcement learning.
• Publications in top‑tier conferences or journals related to machine learning or robotics.
• Proficiency with modern ML frameworks such as PyTorch, TensorFlow, or Jax.

Required skills

• Imitation learning
• Reinforcement learning
• PyTorch
• TensorFlow
• Jax