On-Device ML Infrastructure Engineer (CoreML Runtime), Graphics, Games and Machine Learning
Core
Building the world's most advanced ML graph compilation and runtime system for efficient on-device execution of machine learning models on Apple products.
Role type
Senior IC ML infrastructure engineer (graph compilers and runtimes)
Builds
On-device ML graph compiler, runtime, and kernels
Domain
Consumer electronics / Machine Learning
Deliverable
production ML models
Required skills
C++, Swift, compiler stack (MLIR/LLVM/TVM), operating systems, parallel programming, ML fundamentals (Transformers)
Preferred skills
on-device ML stacks (TFLite/ONNX/ExecuTorch), ML authoring frameworks (PyTorch/TensorFlow/JAX), GPU programming
Technologies
Apple Silicon, Core ML, MLIR, LLVM, TVM
Responsibilities
Architect and maintain the on-device graph compiler, runtime, and kernels; Develop production-critical system software for implementing ML models; Optimize model execution for performance, energy efficiency, and thermal management; Proactively identify and resolve functionality gaps.
