On-Device ML Infrastructure Engineer (CoreML Runtime), Graphics, Games and Machine Learning
Apple· Consumer Technology· Oct 5, 2026 · 22d ago
Cupertino, California, United States
In office
Full-time
Mid
$185K–$325K
Job description
We’re building an end-to-end developer experience for machine learning development that employs Apple’s vertical integration. This allows developers to iterate on model authoring, optimization, transformation, execution, debugging, profiling, and analysis.
This role focuses on the Core ML Runtime for execution on-device. In this role, you will build the world’s most advanced ML graph compilation and runtime system, capable of optimizing and delivering ML models efficiently on Apple products and services.
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, embedding programming, parallel programming.
Sound understanding of ML fundamentals, including common architectures such as Transformers.
Good communication skills, including ability to communicate with multi-functional audiences.
Experience with any on-device ML stack, such as TFLite, ONNX, ExecuTorch, etc.
Experience with any ML authoring framework (PyTorch, TensorFlow, JAX, etc.) is a strong plus.
Experience with accelerators, GPU programming is a strong plus.
Application prep
Use the role’s language and application details while tailoring your materials.
Keywords to mirror
machine learningmlswiftpython
- Platform
- apple
- Effort
- medium