Research Scientist - World Models for Autonomous Driving

ProducePay; United States; Hybrid; USD 161,000 - 269,000; Full time

Benefits offered by this job

Equity incentives

Health and Wellness benefits (medical,

Unlimited Vacation

Flexible hours and Work from Home

Daily drinks, snacks and catered meals

Team building activities

Ongoing evolution of perks

Job summary

Waabi, a leader in Physical AI, is hiring a Research Scientist in World Models to develop and productionize next-generation world-modeling algorithms. You will focus on large-scale models for temporal reasoning, video generation, and multimodal scene understanding to power autonomous driving simulations and planning.

You will collaborate with engineers to deploy models into distributed training and rendering pipelines, publish at top venues, and mentor junior scientists while keeping pace with

Qualifications

• Ph.D. in Computer Vision, ML, Robotics, or related field (or equivalent research experience).
• Expert-level Python and PyTorch (or JAX) with distributed training experience.
• Experience building generative or predictive models of the physical world at scale.
• Strong communication and collaboration in a close-knit team of researchers and engineers.

Responsibilities

• Conduct fundamental and applied research in generative and predictive world-modeling.
• Video generation and prediction.
• Latent diffusion / autoregressive / flow-matching models.
• Multimodal foundation models for driving scenes.
• LLM / VLM / VLA methods for scene understanding, reasoning, and control.
• Generative scenario modeling and controllable simulation.
• Model distillation.
• Collaborate with engineers to integrate models into large-scale, distributed training and rendering pipelines.
• Publish high-impact research at top conferences (CVPR, ECCV, ICCV, NeurIPS, ICLR, ICRA, SIGGRAPH).
• Mentor junior scientists and interns; foster a culture of scientific rigor and rapid experimentation.

Skills

Python

PyTorch

Distributed training

Software engineering fundamentals

Team collaboration

Education

Ph.D. in Computer Vision / ML / Robotics

Tools

JAX

Job description

Waabi, a leader in Physical AI, is hiring a Research Scientist in World Models to develop and productionize next-generation world-modeling algorithms. You will focus on large-scale models for temporal reasoning, video generation, and multimodal scene understanding to power autonomous driving simulations and planning.

You will collaborate with engineers to deploy models into distributed training and rendering pipelines, publish at top venues, and mentor junior scientists while keeping pace with