Senior ML Evaluation & Introspection Engineer

Aurora

Pittsburgh (Allegheny County)

On-site

USD 146,000 - 234,000

Full time

5 days ago
Be an early applicant

Get a reply from this employer

Your resume and cover letter, tailored to exactly what they’re hiring for.

Get past ATS filters

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

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.

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Senior ML Evaluation & Introspection Infra Engineer
Senior ML Evaluation & Introspection Infra Engineer

Aurora • Mountain View (CA)

On-site
USD 162,000 - 260,000
Senior Software Engineer, ML Evaluation & Introspection Infrastructure
Senior Software Engineer, ML Evaluation & Introspection Infrastructure

Aurora • Pittsburgh

On-site
USD 146,000 - 234,000
Senior Software Engineer, ML Evaluation & Introspection Infrastructure
Senior Software Engineer, ML Evaluation & Introspection Infrastructure

Aurora • Mountain View (CA)

On-site
USD 162,000 - 260,000
ML Platform Engineer – Behavior Planning
ML Platform Engineer – Behavior Planning

Tech Talents • Pittsburgh

Hybrid
USD 126,000 - 181,000
ML Platform Engineer Intern: Build Scalable Data Pipelines
ML Platform Engineer Intern: Build Scalable Data Pipelines

Aurora • San Francisco (CA)

On-site
USD 72,000 - 80,000
ML Engineer, Behavior Planning — Autonomy in Motion
ML Engineer, Behavior Planning — Autonomy in Motion

Auror Limited • Mountain View (CA)

Hybrid
USD 139,000 - 223,000
Annual bonus
Equity compensation
Benefits
ML Engineer - Autonomous Driving Behavior Planning
ML Engineer - Autonomous Driving Behavior Planning

Aurora • Mountain View (CA)

Hybrid
USD 139,000 - 223,000
Annual bonus
Equity compensation
Benefits
Senior ML Engineer — Online Maps for Self-Driving
Senior ML Engineer — Online Maps for Self-Driving

Jobright.ai • Mountain View (CA)

On-site
USD 180,000 - 240,000
Senior ML/Infra Engineer for Autonomous Vehicle AI
Senior ML/Infra Engineer for Autonomous Vehicle AI

Overland AI • Seattle (WA)

On-site
USD 120,000 - 300,000
Equity compensation
Best-in-class healthcare
Flexible PTO
+2
Autonomous Driving: ML Engineer, Behavior Planning
Autonomous Driving: ML Engineer, Behavior Planning

Aurora Innovation Inc. • Mountain View (CA)

On-site
USD 139,000 - 223,000
Annual bonus
Equity compensation
Benefits