- Robot type
- Autonomous Vehicle
- Location
- San FranciscoCaliforniaUSA
- Job type
- Software
- Posted
- Apr 2, 2026
- Salary
- $155,000–$269,000 a year
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
Waabi · Software
Senior Network Engineer
Dallas, Texas · Pittsburgh, Pennsylvania
Waabi · Software
Hardware-in-the-Loop (HIL) Software/Firmware Engineer
Pittsburgh, Pennsylvania · San Francisco, California
Waabi · Software
V&V Engineer, Firmware
Pittsburgh, Pennsylvania
Waabi · Software
Staff Systems Engineer, Embedded Safety & Architecture
San Francisco, California · Pittsburgh, Pennsylvania
