Senior/Staff Software Engineer, ML Datasets, Data Pipelines

Posted 3 months ago

Apply Now

Resume Score

Check how well your resume matches this job before you apply.

Sign in to check score

About the role

  • Senior/Staff Software Engineer building data pipelines for Waabi's autonomous driving platform. Collaborating with scientists and engineers to solve complex data challenges in autonomous technology.

Responsibilities

  • Design and implement data pipelines using real-world driving data and Waabi World (our high-fidelity simulator) to train and evaluate deep learning models.
  • Optimize data formats, caching, and dataloading to drive highly efficient ML training and evaluation at scale.
  • Improve data sampling and composition for deep data introspection to track model performance and uncover critical edge-case scenarios.
  • Champion engineering excellence by writing high-quality, well-structured, and rigorously tested code.
  • Help drive project roadmap planning, prioritization, and delivery.

Requirements

  • 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.
  • 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).

Benefits

  • Competitive compensation and equity awards.
  • Health and Wellness benefits encompassing Medical, Dental and Vision coverage (for full-time employees only).
  • Unlimited Vacation.
  • Flexible hours and Work from Home support.
  • Daily drinks, snacks and catered meals (when in office).
  • Regularly scheduled team building activities and social events both on-site, off-site & virtually.
  • As we grow, this list continues to evolve!

Job title

Job type

Full Time

Experience level

Senior

Salary

$148,000 - $260,000 per year

Degree requirement

Bachelor's Degree

Tech skills

AirflowApacheAWSCloudETLHadoopMapReducePythonPyTorchSparkTensorflow

Location requirements

HybridTorontoCanada

Report this job

Found something wrong with the page? Please let us know by submitting a report below.