Senior / Staff Software Engineer, ML Datasets & Data Pipelines
$148,000–$260,000 year
Remote · Toronto, Ontario, Canada or San Francisco, California, United States
Job Summary
As a Senior/Staff Software Engineer on the Autonomy & Algorithms team, you will design and implement scalable ML data pipelines necessary for training and evaluating Waabi’s autonomous driving platform. Responsibilities include optimizing data formats for efficient ML training, improving data sampling for deep introspection, and writing high-quality code. Qualifications include a BS or MS in a relevant field with over 4 years of experience, proficiency in Python, and experience with deep learning frameworks, distributed data processing, and cloud orchestration best practices.
Required Qualifications
- BS or MS in Computer Science, Machine Learning, or a related technical field, with 4+ years of industry experience.
- Proficiency in Python and strong software engineering fundamentals, including experience with deep learning frameworks such as PyTorch, TensorFlow, or JAX.
- Hands-on experience building distributed ETL and data processing pipelines.
- Direct experience managing ML pipelines, including dataset management, dataloading, and optimization.
- Strong understanding of cloud job orchestration, monitoring, and instrumentation best practices.
- A collaborative, open-minded approach with a passion for tackling hard problems in autonomous technology and a strong willingness to mentor others.
Desired Qualifications
- Experience with optimizing large scale distributed training pipelines and/or highly optimized ML inference pipelines.
- Experience with MapReduce (Apache Hadoop/Spark) or orchestration frameworks (Apache Airflow, Apache Beam, Google Cloud Dataflow, AWS Step Functions).
- Experience solving data challenges specific to autonomous driving.
- Familiarity with linear algebra (projections, transforms) and 3D geometry.
- Experience working with multimodal sensor data (e.g., LiDAR, RADAR, camera).
Apply with one swipe on Sorce. We auto-fill applications and apply on your behalf — no cover letters, no 40-minute forms.
Hiring someone like this?
Get your role in front of qualified candidates on Sorce.