On-device ML Performance Engineer, Graphics, Games and Machine Learning
Core
Analyze and optimize latency, memory, power, and numerical correctness of ML models on Apple SoCs, ensuring full utilization of ML accelerators.
Role type
Senior IC on-device ML performance engineer (graphics & ML)
Builds
Optimized ML inference stack for Apple Silicon devices (iPhones, Macs)
Domain
Consumer electronics, on-device machine learning, computer architecture
Deliverable
production ML models
Required skills
ML inference optimization, quantization, computer architecture, CPU/GPU internals, memory management, power analysis, Python, C++, shell scripting
Preferred skills
Apple CoreML, MPS Graph, Metal Performance Shaders, MLX, TFLite, ONNX, ExecuTorch, PyTorch, TensorFlow, JAX, MLIR, LLVM, TVM, Xcode, Swift, Objective-C
Responsibilities
Drive on-device performance analysis of Apple SoCs and ML SW stack; Optimize model conversion, compilation, and inference for performance and energy efficiency; Develop tools and scripts to generate and analyze ML performance data; Collaborate across teams to support hardware and software stack delivery; Present ML performance data to stakeholders
