On-Device ML Graph Compiler & Runtime Engineer

Apple, Inc.; Cupertino (CA); Hybrid; USD 180,000 - 230,000; Full time

Job summary

Apple, Inc. seeks an ML Infrastructure Engineer focused on graph compilers and runtimes for on-device execution. You will help build the world’s advanced ML graph compilation and runtime system, optimizing models to run efficiently on Apple devices while preserving privacy.

You will collaborate across research, software/hardware engineering, and product teams to enable scalable ML workflows on Camera, Siri, Health, Vision, and core experiences.

Qualifications

• Master's degree or equivalent in CS, Engineering, or related field.
• Strong proficiency in C++ or Swift; Python experience is a plus.
• Familiarity with compiler stacks (MLIR/LLVM/TVM).

Responsibilities

• Architect and maintain on-device graph compiler, runtime, and kernels.
• Develop production-grade system software for ML models on Apple Silicon.
• Identify and resolve functionality gaps and optimize execution for performance and energy efficiency.

Skills

C++

Swift

Python

Parallel programming

Education

Master's degree or equivalent

Tools

MLIR/LLVM/TVM

Operating Systems

Embedded systems

Job description

Apple, Inc. seeks an ML Infrastructure Engineer focused on graph compilers and runtimes for on-device execution. You will help build the world’s advanced ML graph compilation and runtime system, optimizing models to run efficiently on Apple devices while preserving privacy.

You will collaborate across research, software/hardware engineering, and product teams to enable scalable ML workflows on Camera, Siri, Health, Vision, and core experiences.