Software Engineer, Data Flywheel Platform

Wayve

Sunnyvale (CA)

On-site

USD 180,000 - 280,000

Full time

3 days ago
Be an early applicant
Application generator

Don’t send a generic resume — generate a resume and cover letter tailored to this exact role.

Get past ATS filters

Job summary

Wayve is hiring in Sunnyvale for a role within the AI Platform that powers data curation, evaluation, and training for our autonomous driving foundation models. You will build scalable data pipelines, evaluation tooling, and training infrastructure used across Science, Autonomy, and Evaluation teams.

Join a world-class team delivering end-to-end aerial-scale ML systems for automakers and mobility partners. This role emphasizes self-serve platforms, reliability, and production-grade software

Qualifications

  • Strong production software engineering experience, particularly in Python services, APIs, and large-scale data processing.
  • Experience with batch and streaming pipelines, orchestration (Flyte, Airflow, Dagster, etc.), and distributed processing (Spark, PySpark, Ray, Databricks).
  • Solid CS fundamentals and several years of production experience; able to design scalable systems.
  • Track record of shipping, testing, observability, and on-call readiness for production systems.

Responsibilities

  • Build systems to turn world-scale driving data into high-signal training data and evaluate/train foundation models.
  • Replace ad-hoc scripts with self-serve, observable products used across teams.
  • Support every model Wayve ships with platform infrastructure and tooling.
  • Collaborate with scientists and engineers to deploy research at automaker scale (Nissan, Stellantis, Uber).
  • Own platform ownership (TC3/TC4) and influence technical direction as the platform matures.

Skills

Production Python
Distributed systems
SQL performance
Observability
Code review

Education

CS degree or equivalent

Job description

About Us

Founded in 2017, Wayve is the leading developer of Embodied AI technology. Our advanced AI software and foundation models enable vehicles to perceive, understand, and navigate any complex environment, enhancing the usability and safety of automated driving systems.

About Us

Founded in 2017, Wayve is the leading developer of Embodied AI technology. Our advanced AI software and foundation models enable vehicles to perceive, understand, and navigate any complex environment, enhancing the usability and safety of automated driving systems.
Our vision is to create autonomy that propels the world forward. Our intelligent, mapless, and hardware-agnostic AI products are designed for automakers, accelerating the transition from assisted to automated driving.
In our fast-paced environment big problems ignite us—we embrace uncertainty, leaning into complex challenges to unlock groundbreaking solutions. We aim high and stay humble in our pursuit of excellence, constantly learning and evolving as we pave the way for a smarter, safer future.
At Wayve, your contributions matter. We value diversity, embrace new perspectives, and foster an inclusive work environment; we back each other to deliver impact.
Make Wayve the experience that defines your career!

About Wayve And The Team

Wayve is building embodied AI for the physical world, starting with autonomous driving. Instead of the hand-engineered, modular stacks that defined the first era of self-driving, we pioneered AV2.0: a single, end-to-end neural network that learns to drive from raw sensor data and generalizes to new cities, vehicles, and conditions. Our foundation models, the GAIA family of generative world models and the LINGO family of vision-language-action models, allow vehicles to perceive, reason, and act in the open world. We have driven zero-shot across hundreds of cities on three continents, and we are now scaling from proving the science to deploying it with leading automakers and mobility partners, including Nissan, Stellantis, and Uber.
This role sits in the AI Platform organization, on the data flywheel that powers every model we ship. Applied Scientists and ML Engineers on the team push the frontier on data curation, enrichment, foundation-model evaluation, and the models themselves. This role builds the platform underneath all of it: the pipelines, infrastructure, and systems that turn world-scale fleet data into high-signal training data, evaluate and train foundation models, and enable every team to run these workflows themselves. As deployment scales, the leverage is enormous: the better the platform, the faster the whole flywheel turns.

The role
  • Build the systems that allow teams to turn world-scale driving data into high-signal training data, and evaluate and train foundation models on it.
  • Replace ad-hoc scripts and manual handoffs with self-serve, observable products used across Science, Autonomy, and Evaluation.
  • Every model Wayve ships runs on this platform: your work compounds across the entire fleet and roadmap.
  • Work shoulder to shoulder with a world-class science and engineering team, with real deployment at global OEM scale (Nissan, Stellantis, Uber).
  • TC3 / TC4 ownership of platform and infrastructure, with room to set technical direction as the platform matures.
What You Will Do
  • Build and scale the data curation and enrichment pipelines that turn world-scale fleet data into high-signal training data: mining and active-learning loops, running model-based enrichments over billions of rows, and ensuring data quality at scale.
  • Build the evaluation infrastructure behind foundation-model progress: harnesses for offline and closed-loop evaluation, metric and benchmark pipelines, and world-model-based evaluation.
  • Build and optimize training and serving infrastructure for large pretrained models: distributed training, batched inference, and large-scale model backfills.
  • Build the data-platform backbone: distributed data processing (Ray Data, Daft, Spark / Databricks), embedding and vector search (turbopuffer, Milvus), lakehouse formats (Lance, Iceberg), dataset versioning, and the enrichment and annotation catalog.
  • Make it self-serve and reliable: turn one-off processes into products that other teams operate themselves, and own testing, observability, and on-call for what you ship.
  • Partner closely with Applied Scientists and ML Engineers to take research from prototype to production at scale.
What We Are Looking For
  • Strong production software engineering, especially production Python (services, APIs, large-scale data processing), and comfort owning and extending large codebases.
  • Large-scale data and distributed-systems experience: batch and streaming pipelines, workflow orchestration (Flyte, Airflow, Dagster, or similar), and distributed processing (Spark / PySpark, Ray, Databricks, or equivalent).
  • Systems design for scale: reliable, observable, high-throughput data or ML systems, with strong SQL and query and performance optimization.
  • A track record of shipping and operating production systems that other teams depend on: testing, code review, observability, and on-call.
  • Strong CS fundamentals and several years of production experience (roughly 6 or more for TC3, more for TC4), or equivalent; a degree in CS or comparable practical experience.
  • Seniority to match the level: takes ambiguous, cross-team problems and drives them to completion, and at TC4 sets technical direction and multiplies the team.
Bonus points
  • ML platform / MLOps: model registration, distributed training, and inference or serving optimization.
  • Enough exposure to foundation models, world models, or ML evaluation to partner deeply with scientists.
  • Embedding and vector search, annotation tooling, or feature and data catalogs.
  • Kubernetes and modern data / lakehouse stacks (Databricks, Lance, Iceberg).
  • Autonomous driving, robotics, or other large-scale sensor-data workflows.

Wayve is committed to creating an inclusive interview experience. If you require any accommodations or adjustments to participate fully in our interview process, please let us know.

At Wayve we're committed to creating a diverse, fair and respectful culture that is inclusive of everyone based on their unique skills and perspectives, and regardless of sex, race, religion or belief, ethnic or national origin, disability, age, citizenship, marital, domestic or civil partnership status, sexual orientation, gender identity, veteran status, pregnancy or related condition (including breastfeeding) or any other basis as protected by applicable law.

DISCLAIMER: We will not ask about marriage or pregnancy, care responsibilities or disabilities in any of our job adverts or interviews. However, we do look to capture information about care responsibilities, and disabilities among other diversity information as part of an optional DEI Monitoring form to help us identify areas of improvement in our hiring process and ensure that the process is inclusive and non-discriminatory.

Get your free, confidential resume review.
or drag and drop your file here.
Similar jobs

Similar jobs worth comparing

Principal Engineer, Model Development Platform
Principal Engineer, Model Development Platform

Icehouseventures • Sunnyvale (CA)

Hybrid
USD 295,000 - 336,000
Full-Stack Software Engineer, Model Development Platform
Full-Stack Software Engineer, Model Development Platform

Lindus Health • Sunnyvale (CA)

On-site
USD 150,000 - 190,000
Applied Scientist / Machine Learning Engineer
Applied Scientist / Machine Learning Engineer

Icehouseventures • Sunnyvale (CA)

Hybrid
USD 311,000 - 370,000
Research Scientist, Wayve Labs
Research Scientist, Wayve Labs

Icehouseventures • Sunnyvale (CA)

Hybrid
USD 140,000 - 180,000
Salary and equity
Onsite chef
Private health insurance
+2
Director of Platform Management for Simulation, Evaluation & Validation
Director of Platform Management for Simulation, Evaluation & Validation

Icehouseventures • Sunnyvale (CA)

On-site
USD 332,000 - 415,000
Principal Engineer, Data & Compute
Principal Engineer, Data & Compute

Wayve • United States

On-site
USD 370,000 - 418,000
Tech Lead, Ingestion Systems
Tech Lead, Ingestion Systems

Wayve • Sunnyvale (CA)

On-site
USD 276,100 - 293,800
Lead Technical Program Manager, Data
Lead Technical Program Manager, Data

Wayve • Sunnyvale (CA)

Hybrid
USD 259,000 - 313,000
Software Engineer, Fleet Management
Software Engineer, Fleet Management

Wayve • Sunnyvale (CA)

Hybrid
USD 177,000 - 221,000
Senior Software Engineer, Fleet Management
Senior Software Engineer, Fleet Management

Wayve • Sunnyvale (CA)

Hybrid
USD 210,000 - 267,000
Competitive equity package
Hybrid work policy