Research Scientist World Models
WaabiVisit Waabi website
Waabi is a Physical AI company developing autonomous-driving technology for commercial trucking and robotaxis.
Toronto, Canada
About Waabi
Founded in 2021 by Raquel Urtasun, Waabi builds a Physical AI platform combining a verifiable end-to-end AI model with neural simulation. Its Waabi Driver and Waabi World technologies support autonomous trucking and robotaxi development.
Skills
Candidate Availability
Required and preferred rules are kept separate and reflect the wording in the original posting.
About the Role
You will research and productionize large-scale world models for dynamic 4D environments. You will develop generative and predictive models, integrate them into distributed pipelines, publish research, mentor others, and follow advances in generative AI and robotics.
Requirements
- PhD in Computer Vision, Machine Learning, Robotics, or a related field, or equivalent research experience
- Expert Python and PyTorch or JAX skills
- Software engineering fundamentals
- Experience with distributed training
- Experience building generative or predictive physical-world models for real-world applications
- Communication skills
Responsibilities
- Conduct research in generative and predictive world modeling
- Develop video, multimodal, LLM, VLM, VLA, and predictive models for driving scenes
- Develop generative scenario models and controllable simulation
- Apply model distillation
- Integrate models into distributed training and rendering pipelines
- Publish research at leading conferences
- Mentor junior scientists and interns
Benefits
- Equity awards
- Medical, dental, and vision coverage for full-time employees
- Unlimited vacation
- Flexible hours
- Work-from-home support
- Daily drinks, snacks, and catered meals when in office
- Team-building activities and social events