On-Device ML Graph Compiler Engineer

Apple; Cupertino (CA); On-site; USD 190,000 - 260,000; Full time

Job summary

Apple is seeking an ML Infrastructure Engineer to advance on-device ML by building graph compilers and runtimes for Apple devices. You will shape the infrastructure enabling scalable ML workflows across Camera, Siri and Vision with tight hardware/software collaboration.

You will craft end-to-end developer experiences for model authoring, optimization, and execution on Apple Silicon, pushing performance and efficiency in constrained environments.

Qualifications

• Masters or equivalent in CS, Engineering or related field.
• Strong proficiency in C++ or Swift, and Python familiarity.
• Experience with compiler stacks (MLIR/LLVM/TVM).
• Knowledge of OS, embedding and parallel programming.
• Solid ML fundamentals and architectures like Transformers.
• Excellent communication across multi-functional teams.

Responsibilities

• Architect and maintain the on-device graph compiler, runtime, and kernels for ML operators.
• Develop production-critical system software for executing ML models on Apple Silicon.
• Identify and resolve functional gaps proactively.
• Optimize model execution for performance, energy, and thermal management.

Skills

C++/Swift

Python

Compiler stacks

Operating Systems

Parallel programming

ML fundamentals

Transformers

Communication

Education

Masters or equivalent

Tools

MLIR

LLVM

TVM

Job description

Apple is seeking an ML Infrastructure Engineer to advance on-device ML by building graph compilers and runtimes for Apple devices. You will shape the infrastructure enabling scalable ML workflows across Camera, Siri and Vision with tight hardware/software collaboration.

You will craft end-to-end developer experiences for model authoring, optimization, and execution on Apple Silicon, pushing performance and efficiency in constrained environments.