Senior / Staff Software Engineer, ML Datasets & Data Pipelines
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 design and implement scalable ML data pipelines using driving and simulation data. You will optimize data formats, caching, and dataloading, improve sampling and data introspection, write tested code, and help plan, prioritize, and deliver projects.
Requirements
- At least 4 years of industry experience
- Python proficiency
- Software engineering fundamentals
- Experience with PyTorch, TensorFlow, or JAX
- Experience building distributed ETL and data-processing pipelines
- Experience managing ML pipelines, dataset management, dataloading, and optimization
- Knowledge of cloud job orchestration, monitoring, and instrumentation
Responsibilities
- Design and implement ML data pipelines for model training and evaluation
- Optimize data formats, caching, and dataloading
- Improve data sampling and composition for model-performance analysis
- Write high-quality, well-structured, rigorously tested code
- Support project roadmap planning, prioritization, and delivery
Benefits
- Equity awards
- Annual performance bonus
- Medical coverage
- Dental coverage
- 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