CoreML On-Device ML Graph Compiler Engineer

Apple Inc.; Cupertino (CA); On-site; USD 181,100 - 318,400; Full time

Benefits offered by this job

Medical and dental coverage

Retirement benefits

Discounted products and free services

Educational reimbursement

Job summary

A leading technology company in Cupertino seeks an On-Device ML Infrastructure Engineer focused on developing advanced ML models for Apple's hardware. The role involves building graph compilers and runtimes, optimizing model execution, and contributing to the AI ecosystem. Ideal candidates will have a Master's in Computer Science, proficient skills in C++ or Swift, and a solid understanding of machine learning principles. This position offers a competitive salary range along with comprehensive benefits, stock programs, and opportunities for career growth.

Qualifications

• Masters or equivalent experience in Computer Sciences, Engineering, or related subject area.
• Highly proficient in C++ or Swift. Familiarity with Python.
• Experience with any compiler stack (MLIR/LLVM/TVM/…).
• Familiarity with Operating Systems and parallel programming.
• Sound understanding of ML fundamentals, including common architectures such as Transformers.

Responsibilities

• Architect and maintain the on‑device graph compiler and runtime for ML.
• Develop production-critical system software for ML models on Apple Silicon.
• Optimize model execution for performance and energy efficiency.
• Proactively identify and resolve functionality gaps.

Skills

C++

Swift

Python

ML fundamentals

Communication skills

Education

Masters in Computer Science or equivalent

Tools

MLIR

LLVM

TFLite

TensorFlow

Job description

A leading technology company in Cupertino seeks an On-Device ML Infrastructure Engineer focused on developing advanced ML models for Apple's hardware. The role involves building graph compilers and runtimes, optimizing model execution, and contributing to the AI ecosystem. Ideal candidates will have a Master's in Computer Science, proficient skills in C++ or Swift, and a solid understanding of machine learning principles. This position offers a competitive salary range along with comprehensive benefits, stock programs, and opportunities for career growth.