Senior/Staff Software Engineer – ML Datasets & Data Pipelines

About the role

As a Senior/Staff Software Engineer embedded within Waabi's Autonomy & Algorithms team, you will build scalable ML data pipelines to train and evaluate the autonomous driving platform, collaborating with world‑renowned scientists and engineers.

Key responsibilities

• Design and implement data pipelines using real‑world driving data and Waabi World 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.

Required profile

• 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.
• Collaborative, open‑minded approach with a passion for tackling hard problems in autonomous technology and a willingness to mentor others.

Required skills

• Python
• PyTorch
• TensorFlow
• JAX
• Distributed ETL
• Data processing pipelines
• ML pipeline management
• Cloud orchestration