Senior Staff Research Engineer Learnable Planner
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
About the Role
You will integrate ML models into a production planning stack through validation, deployment, and monitoring. You will develop simulation interfaces and pipelines, improve planner architectures and representations, work with large datasets, and contribute research findings for conferences and a blog.
Requirements
- Master's or PhD in machine learning, computer science, engineering, or a related field
- Experience with ML-based or classical planning and decision-making techniques
- Understanding of computing fundamentals and code efficiency
- Experience with deep-learning frameworks such as PyTorch
- Proficiency in Python, Rust, C++, and/or CUDA
Responsibilities
- Integrate ML models into the production planning stack
- Develop simulation interfaces and pipelines for testing planning models
- Improve planner architecture and develop representations for end-to-end solutions
- Ensure high-quality, structured, and tested code
- Apply AI, machine learning, computer vision, and self-driving research insights
- Work with large datasets and simulation data
- Contribute research findings for conferences and a blog
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