Research Engineer, World Models

The role

You will design, build and scale generative and predictive world models for autonomous driving. You will work with Research Scientists to turn model prototypes into large-scale training and inference pipelines.

What you will do

• Design, implement and scale generative and predictive world-modeling systems
• Build video generation and prediction models, including latent diffusion, autoregressive and flow-matching models
• Develop multimodal foundation models for driving scenes and LLM/VLM/VLA methods for scene understanding and control
• Build generative scenario modeling and controllable simulation, and model distillation
• Work with Research Scientists to move prototypes into robust distributed training and inference pipelines
• Optimize training and inference for efficiency, speed and reliability on large datasets
• Build large-scale data pipelines to produce high-quality training datasets
• Keep the world model codebase and infrastructure high quality, stable and maintainable
• Follow advances in generative AI, distributed systems and efficient model deployment in robotics

Required qualifications

• Very strong Python and PyTorch (or JAX) skills
• Strong software-engineering fundamentals
• Extensive experience with distributed training and large-scale model deployment
• Master's degree in Computer Vision, Machine Learning, Robotics or a related field, or equivalent industry experience in model development and scaling
• Experience building and deploying generative or predictive models of the physical world
• Experience working in a close-knit team of researchers and engineers, with strong communication in a fast-paced environment

Nice to have

• Experience with infrastructure and tooling for large-scale ML training (e.g., cloud platforms, Kubeflow, Ray)
• Experience with efficient model serving and deployment (e.g., ONNX, TensorRT)
• Publications or research at top ML, CV or robotics conferences (e.g., CVPR, ECCV, NeurIPS)

Benefits

• Competitive compensation and equity awards
• Medical, dental and vision coverage (full-time employees only)
• Unlimited vacation
• Flexible hours and work-from-home support
• Daily drinks, snacks and catered meals when in office
• Regular team building activities and social events on-site, off-site and virtual

Before you apply

• The US salary range applies to US roles; Waabi US Inc. sets salary ranges by company compensation practices

About the company

Waabi, founded by Raquel Urtasun, builds Physical AI for autonomous transportation, including commercial autonomous trucks and robotaxis. The company is backed by and partners with leaders in AI, automotive, logistics and deep tech.