Senior Software Engineer, ML Evaluation & Introspection Infrastructure

Aurora; Pittsburgh (Allegheny County); On-site; USD 146,000 - 234,000; Full time

Job summary

Aurora is hiring a Senior Software Engineer for ML Evaluation & Introspection Infrastructure to join Behavior Analysis. You’ll own core pipelines that compare model versions using large driving datasets and build embeddings, vector search, and model introspection tools.

You will collaborate with ML engineers to define measurements, ensure data quality, and deliver scalable analysis platforms, plus an end-to-end web app and CLI for dataset and run discovery.

Qualifications

• 5+ years of professional software engineering experience building backend, data or ML infrastructure.
• Strong Python, and proficiency with SQL.
• Designed and operated large-scale batch or data pipelines on cloud infrastructure, including backfills, retries and cost management.
• Built systems that measure or evaluate something: metrics, experimentation, evaluation or analytics platforms.
• Experience with distributed compute and container orchestration (e.g. Kubernetes, AWS Batch or similar).

Responsibilities

• Build and scale evaluation pipelines that compare model versions on large driving datasets: offline evals, slice-based metrics, regression detection, and model-vs-model reports.
• Own embedding and vector-search pipelines over driving data, versioned per model, so engineers can search, cluster and compare how any model represents the world.
• Build model introspection infrastructure (attention, attribution, ablations) that turns “why did the model do that?” from a research project into a query.
• Help define how frontier LLMs, VLMs and agents change ML analysis here: triaging, labeling, clustering and summarizing driving scenarios and eval results.
• Work side by side with the ML engineers training these models to decide what to measure, then make those measurements trustworthy, observable and cheap enough to run on every candidate model.
• Build the web app and CLI that engineers and AI agents use to find datasets, training runs, evals and their costs, designed with agents as first-class users.

Skills

Python

SQL

Batch processing

Distributed compute

Kubernetes

Tools

Kubernetes

AWS Batch

Docker

Job description

Who We Are

Aurora’s mission is to deliver the benefits of self-driving technology safely, quickly, and broadly.

Who We Are

Aurora’s mission is to deliver the benefits of self-driving technology safely, quickly, and broadly.

The Aurora Driver will create a new era in mobility and logistics, one that will bring a safer, more efficient, and more accessible future to everyone.

At Aurora, you will tackle massively complex problems alongside other passionate, intelligent individuals, growing as an expert while expanding your knowledge. For the latest news from Aurora, visit aurora.tech or follow us on LinkedIn.

What We Are Looking For

We’re searching for a Senior Software Engineer, ML Evaluation & Introspection Infrastructure to join our Behavior Analysis team, reporting to the team’s engineering manager. Aurora’s trucks already haul freight on Texas highways with no one on board, and every new version of the learned model that drives them has to prove, on real driving data, that it is better and safer than the one before. Behavior Analysis builds the systems that make that call: the evals that compare model versions, the embedding search that makes every driving scenario findable, and the introspection tools that show why a model did what it did. Engineers build more than 100 training and eval datasets a week through our tools. We’re a small team, so each engineer owns whole systems, and we work agent-first: our tools are built for AI agents to drive as easily as engineers do. You’ll own core pieces of that loop and help lead what comes next: putting LLMs, VLMs and agents to work on analysis, so questions that take days to answer today take minutes.

In this role you will

• Build and scale evaluation pipelines that compare model versions on large driving datasets: offline evals, slice-based metrics, regression detection, and model-vs-model reports.
• Own embedding and vector-search pipelines over driving data, versioned per model, so engineers can search, cluster and compare how any model represents the world. This is where you’d start: owning embedding generation and search end to end.
• Build model introspection infrastructure (attention, attribution, ablations) that turns “why did the model do that?” from a research project into a query.
• Help define how frontier LLMs, VLMs and agents change ML analysis here: triaging, labeling, clustering and summarizing driving scenarios and eval results. We already have a prototype using LLMs for data curation, and you’d help decide where it goes next.
• Work side by side with the ML engineers training these models to decide what to measure, then make those measurements trustworthy, observable and cheap enough to run on every candidate model.
• Build the web app and CLI that engineers and AI agents use to find datasets, training runs, evals and their costs, designed with agents as first-class users.

Required Qualifications

• Typically 5+ years of professional software engineering experience building backend, data or ML infrastructure.
• Strong Python, and proficiency with SQL.
• Designed and operated large-scale batch or data pipelines on cloud infrastructure, including backfills, retries and cost management.
• Built systems that measure or evaluate something: metrics, experimentation, evaluation or analytics platforms.
• Experience with distributed compute and container orchestration (e.g. Kubernetes, AWS Batch or similar).

Desirable Qualifications

• ML evaluation infrastructure: eval harnesses, golden datasets, regression detection, A/B model comparison.
• Embeddings and vector search, or model introspection and interpretability tooling.
• Using LLMs, VLMs or agentic workflows to automate analysis, labeling or data curation.
• Autonomous vehicles, robotics or other large-scale sensor data.
• Observability for data systems (metrics, alerting, SLOs), and everyday use of AI coding agents.
• Coming from backend or data engineering and want to learn ML evaluation? Apply anyway: we’ll teach you the ML side.

Working at Aurora

At Aurora, we bring together extraordinarily talented and experienced people united by the strength of our values. We operate with integrity, set outrageous goals, and build a culture where we win together — all without any jerks.

We believe in-person work increases collaboration, empathy, and our ability to lead effectively. As a result, Aurorans are expected to work from the office at least 3 days per week, with some roles requiring more frequent in-office presence, up to 5 days per week, based on operational needs.

Our Careers page provides insight into what it is like to work at Aurora, and you can find all the latest updates in our Newsroom.

Our commitment to safety

At the core of everything we do is our commitment to safety. Building best-in-class self-driving technology will take time, and we believe that each employee at Aurora has a role in contributing to safety, every step of the way. Aurora expects commitment to our safety policies from every employee, and seeks candidates who take an active responsibility, can contribute to building an atmosphere of trust, and invest in the organization’s long-term success by prioritizing working safely, no matter what.

Our commitment to inclusion

Aurora considers candidates without regard to their race, color, religion, national origin, age, sex, gender, gender identity, gender expression, sexual orientation, marital status, pregnancy status, parent or caregiver status, ancestry, political affiliation, veteran and/or military status, physical or mental disability, or any other status protected by federal or state law. Aurora considers qualified applicants with criminal histories, consistent with applicable federal, state, and local law. We are also committed to providing reasonable accommodations for qualified individuals with disabilities and disabled veterans in our job application procedures. If you need assistance or an accommodation due to a disability, you may contact us at [e-mail].

For California applicants, information collected and processed as part of your application and any job applications you choose to submit is subject to Aurora’s California Employment Privacy Policy.

Pay

For roles based in Mountain View, CA and Seattle, WA: The salary range for this position is $162,000 - $260,000 per year.

Aurora’s pay ranges are determined by role, level, and location. Within the range, the successful candidate’s starting base pay will be determined based on factors including job-related skills, experience, qualifications, relevant education or training, and market conditions. These ranges may be modified in the future. The successful candidate will also be eligible for an annual bonus, equity compensation, and benefits.

For roles based in Pittsburgh, PA: The salary range for this position is $146,000 - $234,000 per year.

Aurora’s pay ranges are determined by role, level, and location. Within the range, the successful candidate’s starting base pay will be determined based on factors including job-related skills, experience, qualifications, relevant education or training, and market conditions. These ranges may be modified in the future. The successful candidate will also be eligible for an annual bonus, equity compensation, and benefits.