Apple
On-Device ML Infrastructure Engineer (CoreML Runtime), Graphics, Games and Machine Learning
$185K–$325K
Cupertino, California · Full-time · Posted Oct 5, 2026
- Education
- Master’s degree or equivalent experience in Computer Science, Engineering, or a related subject area.
About the job
Build and optimize the Core ML on-device graph compiler, runtime, and kernels so machine-learning models execute efficiently on Apple devices.
What you'll do
- Architect and maintain the on-device graph compiler, runtime, and ML operator kernels.
- Develop production-critical system software for ML models on Apple Silicon and address functionality gaps.
- Optimize model execution for performance, energy efficiency, and thermal management.
What they ask for
- Master’s degree or equivalent experience in Computer Science, Engineering, or a related subject area.
- High proficiency in C++ or Swift; familiarity with Python.
- Experience with a compiler stack such as MLIR, LLVM, or TVM.
- Familiarity with operating systems, embedded programming, and parallel programming; sound understanding of ML fundamentals, including Transformers.
Nice to have
- Experience with an on-device ML stack such as TFLite, ONNX, or ExecuTorch.
- Experience with an ML authoring framework such as PyTorch, TensorFlow, or JAX; accelerator or GPU programming experience.
Pay
$184,700 and $324,800 USD, base salary
Tools
C++, Swift, Python, MLIR, LLVM, TVM, TFLite, ONNX
AI focus
Develop on-device machine-learning compilation and runtime infrastructure, optimizing ML models for efficient execution on Apple devices.
Model-written summary · the original posting is authoritative