← All roles

Research Engineer, World Models

Waabi· Software – Autonomy & Algorithms
Apply on company site
Hybrid (Toronto, SF, Pittsburgh, Remote)$155k - $269kFull-timePosted 6mo ago

About the role

Research Engineer in Waabi’s World Models team focused 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 to power Waabi World’s modeling of future evolution and controllable simulation.

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.
  • Build large scale data pipelines to build high quality datasets for training
  • Ensure the quality, stability, and maintainability of the world model codebase and infrastructure.
  • Stay on top of emerging advances in generative AI, distributed systems, and efficient model deployment in robotics.

Qualifications

  • Strong software engineering and implementation: very strong Python & PyTorch (or JAX) skills; strong software-engineering fundamentals; extensive experience with distributed training and large-scale model deployment.
  • Demonstrated technical impact: Master’s degree in Computer Vision, Machine Learning, Robotics, or related field, or equivalent industry experience in model development and scaling.
  • Expert domain knowledge: built and deployed generative or predictive models of the physical world, focusing on scale, efficiency, and robustness for real-world applications.
  • Team player: experience in a close-knit team of researchers and engineers with strong communication to deliver successful projects in a fast-paced environment.

Skills

Required:PythonPyTorchDistributed training
Nice to have:Video generation and predictionMultimodal modelsLLM/VLM/VLA methods

Benefits

  • Competitive compensation and equity awards.
  • Health and Wellness benefits encompassing Medical, Dental and Vision coverage (for full-time employees only).
  • Unlimited Vacation.
  • Flexible hours and Work from Home support.
  • Daily drinks, snacks and catered meals (when in office).
  • Regularly scheduled team building activities and social events both on-site, off-site & virtually.
Last updated Oct 21, 2026. Interested? Apply on the company site.

About Waabi

pioneering Physical AI, starting with autonomous trucks.

Waabi is pioneering Physical AI, starting with autonomous trucks. We developed a next-generation approach leveraging an end-to-end interpretable and verifiable AI model that’s powered by the industry's most realistic neural simulator. This dramatically reduces the development time and resources needed to bring self-driving vehicles to public roads — safely and at scale.

Autonomous trucksPhysical AIneural simulatorAI modelself-driving vehicles
Company profileWebsite

More open roles at Waabi