Senior Staff ML Onboard Optimization Engineer
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 work with autonomy and algorithm engineers to scale self-driving systems. You will expand model deployment to GPUs and embedded platforms, optimize truck models with TensorRT and modelopt, create and benchmark CUDA kernels, and profile runtime and memory to identify performance bottlenecks.
Requirements
- MS/PhD or a bachelor's degree with at least 6 years of industry experience in Computer Science, Robotics, or a similar technical field
- Proficiency in Python, C++, or Rust
- Experience with PyTorch
- Experience profiling CPU and GPU code with PyTorch Profiler and NVIDIA Nsight
- Experience with NVIDIA embedded platforms such as Jetson or Thor
Responsibilities
- Collaborate with autonomy and algorithm engineers to scale self-driving systems
- Expand model deployment to GPUs and embedded systems
- Optimize models running on trucks using TensorRT and modelopt
- Create and benchmark CUDA kernels for inference
- Profile model runtime and memory to identify performance bottlenecks
Benefits
- Equity awards
- Medical, dental, and vision coverage
- Unlimited vacation
- Flexible hours
- Work-from-home support
- Daily drinks, snacks, and catered meals when in office
- Team-building activities and social events
- On-site gym, games room, collaborative spaces, and patio