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