On-Device ML Compiler Engineer, Model Compilation, Graphics, Games and Machine Learning

Apple Inc

Quick summary

Work type: On-site
Location: Cupertino, CA
Salary: $147,400–$272,100 / yr
Posted: 56 days ago
Nearby: 99+ roles within 25 mi

Market check

Salary context

Competitive pay

How this pay compares to similar roles

Similar $214k

This role $210k

$132k most similar roles pay here $287k

This role pays more than 54% of similar roles. Most pay $190,500–$236,900 — the shaded band above. At the midpoint, this role pays about $210k versus about $214k for comparable roles.

Based on 239 similar postings.

Employer

About Apple Inc

Apple Inc. is a multinational technology company known for designing and manufacturing consumer electronics, software, and online services, including the iPhone, Mac, iPad, and App Store. Industry: Consumer Electronics & Software

Apple Inc currently has 1723 open roles on FindRole.

Listed pay typically runs $162,500–$272,100 across 1398 roles with salary data.

Most-posted roles

View all roles at Apple Inc

At a glance

TL;DR · On-Device ML Compiler Engineer, Model Compilation, Graphics, Games and Machine Learning

Apply Now Log in to save

As an On-Device Machine Learning Compiler Engineer at Apple, you will join a dynamic team dedicated to transforming cutting-edge AI research into practical applications that run efficiently on Apple devices. Your primary responsibilities include enhancing the MLIR-based compiler stack to optimize runtime performance across various hardware components such as the neural engine, GPU, and CPU. You will collaborate closely with model authoring, runtime, and performance teams to ensure seamless integration of innovative architectures into embedded systems. Key tasks involve proposing upstream changes in MLIR to support new features and workflows, owning critical parts of the compiler stack for heterogeneous computing, and working across hardware, software, and performance teams to leverage the latest advancements in hardware capabilities. This role requires expertise in C++, familiarity with PyTorch or similar frameworks, and a passion for machine learning and system software engineering.

Skills

MLIR C++ PyTorch Swift GPU CPU Neural_Engine CI/CD

What you'll do

Propose changes to MLIR-based compiler to enhance runtime performance on Apple hardware.
Own core components of the compiler stack enabling efficient cross-device computation in the Apple ecosystem.
Collaborate with teams across the company to optimize and accelerate ML model execution.
Demonstrate hardware capabilities through improvements in the MLIR-based compiler infrastructure.
Work on upstream changes in MLIR to support new features and workflows for optimal performance.

What we're looking for

3-5 years of experience working on MLIR-based compilers.
Proficiency in C++ and familiarity with PyTorch or similar training frameworks.
Understanding of common machine learning model architectures, operations, and execution schemes.
Experience with Swift is beneficial but not required.
Knowledge of programming paradigms for GPU, CPU, and Neural Engine.
Ability to write kernels for ML model execution.