← All roles

Research Scientist, World Models

Waabi· Software – Autonomy & Algorithms
Apply on company site
Hybrid (Toronto, ON / San Francisco, CA / Pittsburgh, PA / Remote US)$155k - $269kFull-timePosted 6mo ago

About the role

Research Scientist in World Models developing large-scale world models for temporal reasoning and generation to power Waabi World, including video models, multimodal generative models, LLM/VLM/VLA models, and predictive models of traffic participants and scenes.

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
  • Stay on top of emerging advances in generative AI, differentiable rendering, knowledge distillation/compression, and robotics

Qualifications

  • Ph.D. in Computer Vision, Machine Learning, Robotics, or a related field or equivalent research experience pushing the boundaries of a technical field
  • Expert-level Python & PyTorch (or JAX) skills; strong software-engineering fundamentals and experience with distributed training
  • Built generative or predictive models of the physical world with scale and efficiency for real-world applications
  • Strong communication and ability to work in a close-knit team of researchers and engineers

Skills

Required:Generative and predictive world-modelingMultimodal foundation models (driving scenes)
Nice to have:LLM / VLM / VLA methods

Benefits

  • Competitive compensation and equity awards
  • Health and Wellness benefits encompassing Medical, Dental and Vision (full-time)
  • 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 & virtual)
  • Continued evolution of benefits as the company grows
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