Skip to content

Jobs

Waabi

Robot type
Autonomous Vehicle
Location
San FranciscoCaliforniaUSA
Job type
Software
Posted
Apr 2, 2026
Salary
$155,000–$269,000 a year
Full-time

Research Engineer, World Models

Job description

This role focuses on large-scale world models for temporal reasoning and generation, including video models, multimodal generative models, LLM/VLM/VLA models, and predictive models of traffic participants and scenes. Your work will directly power Waabi World’s ability to model future evolution, synthesize realistic safety-critical scenarios, and provide rich generative priors for downstream planning, testing, and training.

Waabi’s autonomous-driving stack is powered by Waabi World, which delivers realistic, scalable, controllable, and efficient simulation. As a Research Engineer in the World Models team, you will develop algorithms and productionize the next generation of World Models that…

Job responsibilities

  • Design, implement, and scale state-of-the-art generative and predictive world-modeling systems: 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 closely with Research Scientists to translate cutting-edge model prototypes into robust, large-scale, distributed training and inference pipelines.
  • Optimize model training and inference for efficiency, speed, and reliability on large-scale datasets.

Similar jobs