Senior/Staff Software Engineer – ML Datasets & Data Pipelines

Job overview

The Senior/Staff Software Engineer – ML Datasets & Data Pipelines role is hybrid in San Francisco, CA, with a U.S. annual salary range of $148,000–$260,000. Waabi builds Physical AI technology for autonomous trucks and robotaxis, and this role is embedded in its Autonomy & Algorithms team. The engineer builds scalable machine-learning data pipelines used to train and evaluate the autonomous driving platform. The work improves data efficiency and model evaluation, helping the team identify performance issues and edge cases as it develops driverless vehicle technology.

What you'll do

• Design and implement data pipelines using real-world driving data and the Waabi World simulator
• optimize data formats, caching, and data loading
• improve data sampling and composition for model performance analysis and edge-case discovery
• write tested, well-structured code
• contribute to roadmap planning, prioritization, and delivery.

What we're looking for

Skills & competencies

senior software engineer

staff software engineer

machine learning

ml datasets

data pipelines

python

pytorch

tensorflow

autonomous driving

hybrid

san francisco ca

waabi

$148k–$260k/year

equity awards

Work arrangement

• Weekend coverage required

Benefits & perks

• Equity awards
• medical, dental and vision coverage for full-time employees
• unlimited vacation
• flexible hours
• work-from-home support
• office drinks, snacks, and catered meals.

Why this role

The posting lists equity awards, medical, dental and vision coverage for full-time employees, unlimited vacation, flexible hours, work-from-home support, and office meals and snacks. Additional compensation may include equity incentives and an annual performance bonus.

About the employer

Waabi is hiring for this role. Industry: Computing Infrastructure Providers, Data Processing, Web Hosting, and Related Services. Sector: 51.