Machine Learning Engineer, Performance Tooling at Wayve | CA, US | Rezi

Machine Learning Engineer, Performance Tooling at Wayve

Machine Learning Engineer, Performance Tooling

Wayve · CA, US

6 days ago

Machine Learning Engineer, Performance Tooling

Wayve · CA, US

6 days ago
Resume preview

Impress employers and recruiters.
Choose from hundreds of resume examples.

Target Resume Now
Resume preview

Tailor your resume to this Machine Learning Engineer, Performance Tooling role.

Rezi rewrites your resume against Wayve's job description. Free.

Resume score gauge reading 58 out of 100

Don't guess if your resume is good enough.

See how it scores against the Machine Learning Engineer, Performance Tooling posting at Wayve — free, in seconds.

About the Role

You will join the AI Performance Tooling team, focusing on making model training and inference faster, more efficient, and predictable across cloud and embedded hardware. Your mission is to enable data-driven AI performance decisions by building tools that reason across the AI stack, transforming profiling data into clear insights on performance bottlenecks and potential improvements. You will collaborate with various engineering teams to provide cross-stack analysis and recommendations.

Responsibilities

  • Design and build scalable, self-service performance tools for models, hardware targets, and development workflows.
  • Define standards for measuring and predicting AI performance at Wayve.
  • Analyze platform performance against theoretical peaks, identifying efficiency losses at the layer and op level.
  • Predict the performance impact (latency, memory, utilization, compute cost) of model or recipe changes before execution.
  • Manage monitoring and regression alerting for model builds and training runs.
  • Collaborate with training and runtime engineers to establish performance targets and present data-driven justifications.

Requirements

  • Deep, hands-on performance engineering experience in complex systems, including profiling, roofline analysis, latency/throughput optimization, and root-cause analysis.
  • Proven track record of owning a tool or service from design through adoption.
  • Strong Python skills with experience in profiling and instrumenting large production codebases.
  • Hands-on experience developing deep learning models using PyTorch.
  • Proficiency in data analysis to derive defensible conclusions from measurements.
  • Ability to translate ambiguous performance questions into measurable problems and prioritize effectively.
  • Quantitative communication skills to influence other teams' priorities.

Skills

  • Performance Engineering
  • Profiling
  • Roofline Analysis
  • Latency Optimization
  • Throughput Optimization
  • Root-Cause Analysis
  • Python
  • PyTorch
  • Data Analysis
  • Quantitative Communication

Location

  • In-office
  • Hybrid

Work Type

  • Full-time
  • Hybrid

Experience Level

  • Senior

About the Company

  • Founded in 2017, Wayve is a leader in Embodied AI technology, developing advanced AI software and foundation models for vehicles to perceive, understand, and navigate complex environments.
  • Wayve's vision is to create autonomy that propels the world forward with mapless, hardware-agnostic AI products designed to accelerate the transition to automated driving.
  • The company fosters a fast-paced environment that embraces uncertainty and complex challenges, encouraging learning, evolution, and impact.
  • Wayve values diversity, new perspectives, and an inclusive work environment where contributions matter.

Equal Opportunity

  • Wayve is committed to creating an inclusive interview experience and encourages applications from individuals who may not meet every listed requirement but are passionate about self-driving cars.
  • The company is committed to fostering a diverse, fair, and respectful culture that is inclusive of everyone, regardless of protected characteristics.
  • For US candidates: E-Verify Notice and Participation and Right to Work information is available.