Software Engineer, Data Flywheel Platform

Wayve • Sunnyvale, CA

3d

About The Position

Wayve is building embodied AI for the physical world, starting with autonomous driving. Instead of hand-engineered, modular stacks, Wayve 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.

Requirements

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

Nice To Haves

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

Responsibilities

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

Benefits

• 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!
• We are 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.
• We understand that everyone has a unique set of skills and experiences and that not everyone will meet all of the requirements listed above. If you’re passionate about self-driving cars and think you have what it takes to make a positive impact on the world, we encourage you to apply.
• 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.

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Education Level: No Education Listed