On-Device ML Performance Engineer: Graphics & Inference

Apple Inc.; Seattle (WA); On-site; USD 142,000 - 263,000; Full time

Benefits offered by this job

Employee stock programs

Restricted stock units

Stock Purchase Plan

Medical and dental coverage

Retirement benefits

Education reimbursement

Job summary

Apple Inc. is seeking an On-device ML Performance Engineer to optimize the performance of the latest ML models on iPhone, iPad and Mac hardware. You will analyze CPU/GPU/ENE drivers and memory usage while integrating with core ML frameworks and the ML software stack.

The role requires deep knowledge of computer architecture, ML inference, and on-device optimization, with scripting and tooling to extract performance metrics. Seattle-based position.”

Qualifications

• Experience with ML inference, quantization, performance and accuracy.
• Familiarity with popular ML architectures (e.g., LLMs, Diffusion, CNNs).
• Interest in on-device inference and HW/software stack optimization.

Responsibilities

• Analyze on-device performance of Apple SoCs and ML software stack.
• Optimize model conversion, compilation and on-device inference for energy and memory efficiency.
• Develop tools and scripts to generate and analyze performance data.
• Collaborate across teams to support on-device ML hardware/software delivery.
• Present ML performance data to internal and external stakeholders.

Skills

ML inference

Quantization

Model architectures

Python/C++

Linux/macOS

Communication

Education

Masters or PhD in CS or related

Tools

CoreML

MPS Graph

Metal Performance Shaders

MLX

Job description

Apple Inc. is seeking an On-device ML Performance Engineer to optimize the performance of the latest ML models on iPhone, iPad and Mac hardware. You will analyze CPU/GPU/ENE drivers and memory usage while integrating with core ML frameworks and the ML software stack.

The role requires deep knowledge of computer architecture, ML inference, and on-device optimization, with scripting and tooling to extract performance metrics. Seattle-based position.”