Drimva

Waabi

Research Scientist, World Models

$155K–$269K

Toronto, ON +3 · Hybrid · Full-time · Posted Mar 27, 2026

Location
Toronto, ON
San Francisco, CA
Pittsburgh, PA
Remote US
Education
Ph.D. in Computer Vision, Machine Learning, Robotics, or a related field or equivalent research experience.

About the job

Develop and productionize large-scale generative and predictive world models for autonomous-driving simulation, temporal reasoning, and realistic scenario generation.

What you'll do

  • Research and build video, multimodal, and predictive models for driving scenes and controllable simulation.
  • Integrate models into distributed training and rendering pipelines with engineering teams.
  • Publish research and mentor junior scientists and interns.

What they ask for

  • Ph.D. in Computer Vision, Machine Learning, Robotics, or a related field, or equivalent research experience.
  • Expert-level Python and PyTorch (or JAX) skills, strong software-engineering fundamentals, and distributed-training experience.
  • Experience building scalable and efficient generative or predictive models of the physical world for real-world applications.
  • Strong communication and experience collaborating with researchers and engineers.

Nice to have

  • Ability to translate research into production-quality code with measurable product impact.
  • First-author publications in top-tier venues on relevant topics.

Pay

$155,000 - $269,000 USD (base salary)

Tools

Python, PyTorch, JAX

AI focus

Research and productionize world models, including video generation and prediction, latent diffusion, autoregressive and flow-matching models, multimodal foundation models, LLM/VLM/VLA methods, scenario modeling, and model distillation.

Model-written summary · the original posting is authoritative

First seen Oct 10, 2026 · Last verified Nov 5, 2026 (UTC)