Waabi
Senior / Staff Software Engineer, ML Datasets & Data Pipelines
$148K–$260K
Toronto, ON +2 · Hybrid · Staff · Full-time · Posted Mar 26, 2026
- Location
- Toronto, ON
San Francisco, CA
Remote US & Canada - Experience
- 4+ years of industry experience.
- Education
- BS or MS in Computer Science, Machine Learning, or a related technical field.
About the job
Build scalable ML data pipelines to train and evaluate Waabi’s autonomous-driving platform using real-world and simulated driving data.
What you'll do
- Design and implement pipelines for training and evaluating deep learning models.
- Optimize data formats, caching, dataloading, sampling, and dataset composition for ML workloads.
- Write rigorously tested code and help plan, prioritize, and deliver projects.
What they ask for
- BS or MS in Computer Science, Machine Learning, or a related technical field.
- Proficiency in Python and strong software engineering fundamentals; experience with deep learning frameworks.
- Hands-on experience building distributed ETL and data processing pipelines.
- Direct experience managing ML pipelines, including dataset management, dataloading, and optimization; understanding of cloud job orchestration, monitoring, and instrumentation.
Nice to have
- Experience optimizing large-scale distributed training or ML inference pipelines; experience with MapReduce or orchestration frameworks.
- Experience with autonomous-driving data, linear algebra and 3D geometry, or multimodal sensor data.
Pay
$148,000 - $260,000 USD (base salary)
Tools
Python, PyTorch, TensorFlow, JAX, Apache Hadoop, Apache Spark, Apache Airflow, Apache Beam
AI focus
Build and optimize ML data pipelines and datasets used to train and evaluate deep learning models for autonomous driving.
Model-written summary · the original posting is authoritative