Senior / Staff Software Engineer, ML Datasets & Data Pipelines

Job summary

Responsibilities

With offices in Toronto, San Francisco, Dallas, and Pittsburgh, Waabi is growing quickly and looking for diverse, innovative and collaborative candidates who want to impact the world in a positive way. To learn more visit: www.waabi.ai

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

What they're looking for

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

Tech stack

• Python
• PyTorch
• TensorFlow
• JAX
• Spark
• Airflow
• GCP
• AWS