On-Device ML Compiler Engineer

Apple Inc

Confirmed live yesterday High trust

Quick summary

Work type
On-site
Location
Cupertino, CA
Salary
$150,400–$277,600 / yr
Posted
58 days ago
Freshness
Confirmed live yesterday

Market check

Salary context

Competitive pay

How this pay compares to similar roles

Similar $220k
This role $214k
$135k most similar roles pay here $293k

This role pays more than 58% of similar roles. Most pay $185,312–$254,750 — the shaded band above. At the midpoint, this role pays about $214k versus about $220k for comparable roles.

Based on 240 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 1984 open roles on FindRole.

Listed pay typically runs $175,000–$277,600 across 1590 roles with salary data.

Most-posted roles

View all roles at Apple Inc

At a glance

TL;DR · On-Device ML Compiler Engineer

The On-Device ML Compiler Engineer joins the On-Device Machine Learning team to build essential infrastructure for running advanced models locally across various devices. This role focuses on model compilation and core runtime execution, where you will work with hardware, software, and performance teams to optimize execution by leveraging the latest features in the OS and drivers. You will own pieces of the compiler stack to enable heterogeneous compute across the ecosystem, from resource-constrained devices like Apple Watch to high-end Macs. The role involves proposing upstream changes to an MLIR-based compiler stack to target the neural engine, GPU, and CPU. Required skills include C++, PyTorch or related training frameworks, and experience with MLIR-based compilers. Preferred qualifications include Swift, writing kernels for model execution, and knowledge of programming paradigms for the GPU, CPU, and Neural Engine.

What you'll do

  • Develop and optimize MLIR-based compiler stacks to target the Neural Engine, GPU, and CPU.
  • Propose upstream changes in MLIR to support new hardware features and workflows.
  • Improve runtime performance by identifying and implementing capabilities of Apple silicon.
  • Own core components of the compiler stack that enable heterogeneous compute across all Apple devices.
  • Optimize machine learning model execution for a range of devices from Apple Watch to high-end Macs.
  • Collaborate with hardware, software, and performance teams to leverage latest OS and driver features.
  • Build infrastructure for model compression, acceleration, and efficient runtime execution.

What we're looking for

  • Experience of 3-5 years working on MLIR-based compilers.
  • Familiarity with C++.
  • Familiarity with PyTorch or related training frameworks.
  • Familiarity with common ML model architectures, execution schemes, and operations.
  • Familiarity with Swift.
  • Familiarity with programming paradigms for the GPU, CPU, and Neural Engine.
  • Familiarity with writing kernels for ML model execution.

More like this

Similar roles

On-Device ML Compiler Engineer

Apple Inc

Cupertino, CA 8 days ago $184,700$324,800
MLIR C++ PyTorch Swift GPU Neural Engine Compiler Optimization Machine Learning Model Compression System Software Engineering Profiling Debugging Benchmarking

Machine Learning Compiler Engineer

Apple Inc

Sunnyvale, CA 36 days ago $150,400$277,600
C++ LLVM MLIR Compiler Design Intermediate Representation JIT) compilation Neural Network Inference Multi-threading Program Analysis Register Allocation Code Generation Deep Learning Frameworks
3+ yrs exp

GPU ML Engineer

Apple Inc

Cupertino, CA 23 days ago $184,700$324,800
Metal Performance Shaders GPU Programming Machine Learning Linear Algebra Computer Vision Parallel Programming Computer Architecture Image Processing Numerical Methods
2+ yrs exp