On-Device ML Infrastructure Engineer (CoreML Runtime), Graphics, Games and Machine Learning

Apple Inc

Quick summary

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

Market check

Salary context

Above market

How this pay compares to similar roles

Similar $213k

This role $250k

$144k most similar roles pay here $337k

This role pays more than 86% of similar roles. Most pay $188,260–$237,875 — the shaded band above. At the midpoint, this role pays about $250k versus about $213k 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 Infrastructure Engineer (CoreML Runtime), Graphics, Games and Machine Learning

Apply Now Log in to save

As an On-Device ML Infrastructure Engineer at Apple’s Graphics, Games and Machine Learning team, you will play a pivotal role in advancing the Core ML Runtime for execution on Apple devices. Your primary responsibilities include architecting and maintaining the graph compiler, runtime, and kernels that enable efficient delivery of machine learning operators on resource-constrained hardware. You will develop production-critical system software to implement ML models on Apple Silicon while optimizing model execution for performance, energy efficiency, and thermal management. Proficiency in C++ or Swift is essential, along with experience in compiler stacks like MLIR/LLVM/TVM, and familiarity with operating systems and parallel programming. This role demands a deep understanding of machine learning fundamentals and the ability to work creatively within a fast-paced environment to address technical challenges and enhance intelligent experiences across Apple’s ecosystem.

Skills

C++ Swift Python MLIR LLVM TVM TFLite ONNX ExecuTorch PyTorch TensorFlow JAX CUDA OpenCL CI/CD

What you'll do

Architect and maintain the on-device graph compiler for efficient ML operator delivery.
Develop production-critical system software to implement ML models on Apple Silicon.
Identify and resolve functionality gaps in the Core ML Runtime.
Optimize model execution for performance, energy efficiency, and thermal management.
Build comprehensive benchmarking and debugging toolchains for ML development.

What we're looking for

Master’s degree in Computer Science or equivalent experience.
Proficiency in C++ and familiarity with Swift and Python.
Experience with compiler stacks like MLIR, LLVM, or TVM.
Understanding of machine learning fundamentals and architectures.
Familiarity with operating systems, embedded programming, and parallel programming.
Good communication skills for multi-functional audiences.