← All roles

Senior / Staff Software Engineer, ML Datasets & Data Pipelines

Waabi· Software – Autonomy & Algorithms
Apply on company site
Hybrid (Toronto, ON / San Francisco, CA / Remote US & Canada)$148k - $260kFull-timePosted 7mo ago

About the role

Senior/Staff Software Engineer embedded within the Autonomy & Algorithms team, building scalable ML data pipelines to train and evaluate Waabi’s autonomous driving platform, working with world-class scientists and engineers to solve complex data challenges.

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.

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.

Skills

Required:PythonDeep learning frameworksDistributed ETL and data processingML pipelines / dataset managementCloud job orchestration and monitoring

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.
Last updated Oct 21, 2026. Interested? Apply on the company site.

About Waabi

pioneering Physical AI, starting with autonomous trucks.

Waabi is pioneering Physical AI, starting with autonomous trucks. We developed a next-generation approach leveraging an end-to-end interpretable and verifiable AI model that’s powered by the industry's most realistic neural simulator. This dramatically reduces the development time and resources needed to bring self-driving vehicles to public roads — safely and at scale.

Autonomous trucksPhysical AIneural simulatorAI modelself-driving vehicles
Company profileWebsite

More open roles at Waabi