Hiring.Camp

Senior / Staff Software Engineer, ML Datasets & Data Pipelines

Waabi

·

Mar 26, 2026

Location
Toronto, ON · San Francisco, CA · Remote US & Canada
Workplace
Hybrid, Remote
Type
Full-time
Department
Software
Seniority
Senior
Source
Lever

Description

Waabi, founded by AI visionary Raquel Urtasun, is the leader in Physical AI. With a world-class team, we're unlocking the next era of autonomous transportation with technology that's powering commercial autonomous trucks and robotaxis. Waabi is backed by and partners with world leaders in AI, automotive, logistics, and deep tech.

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

As a Senior/Staff Software Engineer embedded within our Autonomy & Algorithms team, you will build the scalable ML data pipelines necessary to train and evaluate Waabi’s autonomous driving platform. Working closely with world-renowned scientists and engineers, you will solve complex data challenges to accelerate our launch of fully driverless vehicles.

 

You Will..

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

 

Bonus Points:

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

Skills

PythonAWSMachine LearningDeep LearningTensorFlowPyTorchSparkHadoopAirflowETL

Similar Jobs

30

Builder - Senior/Staff Software Engineer

Reevo · Santa Clara

Today

Senior Staff Equipment Mechanical Design Engineer

Amperesand · Reno, Nevada, United States +1

Today

Senior Staff Engineer, CAE Infrastructure & R&D Software

Samsungsemiconductor · San Jose, California, United States · Onsite

Today

Senior Staff Mechanical Design Engineer

Amperesand · Reno, Nevada, United States; San Francisco, California, United States +2

Today

Senior/Staff Data Scientist

Sovrn · Boulder, Colorado +1

Today

Senior Staff Accountant, Journal Entry

LifePoint Health · TN, United States, US · Remote

Today

Senior/Staff Engineer - Exchange Middle Platform

OKX · Singapore, Singapore

Today

Senior Staff Engineer, TLM Banking Reconciliation

Nagarro · Remote, India · Remote

Today

Senior Staff Technical Program Manager

Crusoe · San Francisco, CA - US · Onsite

Yesterday

Senior Staff Engineer - Infrastructure Engineering (Linux), OS Team

Geico · MD Bethesda Office, United States of America +3 · Hybrid

Yesterday

CPU RTL Design Engineer (Senior/Staff) - Cambridge, UK

Qualcomm · Cambridge, England,GB, GB

Yesterday

SENIOR STAFF ANALYST (Housing and Homelessness)

Fresno County · Fresno County, CA, CA, US

Yesterday

Staff or Senior Staff Attorney , Workers' Compensation

Thehartford · Brea - Olen Pointe Dr, United States of America +2

Yesterday

Strip Test Senior/Staff Engineer

Nexperia · Seremban, Malaysia

Yesterday

Senior/Staff Product Engineer (Optical Transceiver)

Lumilens · Singapore

Yesterday

Senior/Staff Process Engineer (Wire Bond)

Lumilens · Singapore

Yesterday

Senior/Staff Process Engineer (Flip Chip)

Lumilens · Singapore

Yesterday

Senior/Staff Process Engineer (Die Attach)

Lumilens · Singapore

Yesterday

Senior/Staff Process Engineer (Assembly)

Lumilens · Singapore

Yesterday

Senior/Staff Process Engineer (Active Alignment)

Lumilens · Singapore

Yesterday

Senior/Staff Industrial Engineer

Lumilens · Singapore

Yesterday

Senior/ Staff Engineer, Front End Labor Productivity

Micron Technology · SG

Yesterday

Senior/ Staff Engineer, Front End Labor Productivity

Micron · Fab 10A, Singapore

Yesterday

Senior Staff Software Engineer

Kaleris · Chennai, India +1 · Remote

Yesterday

Senior/Staff Analytics Engineer

Setpoint · Austin or New York (Hybrid) +1 · Hybrid

Yesterday

Software Engineer Staff/Senior Staff (TS Clearance)

Solerity · Denver, CO · Remote, Hybrid, Onsite

Yesterday

Senior/Staff Software Engineer, Distributed Systems

Atomic Machines · Emeryville, California or Santa Clara, California +1

Yesterday

Senior/Staff Software Engineer, Manufacturing Systems

Atomic Machines · Emeryville, California or Santa Clara, California +1

Yesterday

Senior Staff Engineer, Autonomy and Motion Planning (R5538)

Shieldai · London · Onsite

Yesterday

Senior Staff Engineer, Autonomy Software Integration & Test (R5539)

Shieldai · London · Onsite

Yesterday
Remote Senior / Staff Software Engineer, ML Datasets & Data Pipelines at Waabi | Hiring.Camp