← All roles

On-device ML Performance Engineer, Graphics, Games and Machine Learning

Apple· On-device ML Performance Team
Apply on company site
Cupertino, California$150.4k - $277.6kFull-timePosted 1mo ago

About the role

Engineer focusing on analyzing and optimizing the performance, power, and memory of on-device ML models running on Apple Silicon, working with PyTorch/ML frameworks, developing tools, and collaborating across teams to enable high-performance on-device ML inference on iPhone and Mac.

Responsibilities

  • Driving the on-device performance analysis of Apple SoC’s and ML SW stack across a wide range of Apple internal or open-source ML models
  • Using innovative ways to optimize model conversion, compilation and on-device inference for Apple SoC’s, achieving ambitious goals for performance, memory, and energy efficiency
  • Developing tools and scripts to generate and analyze ML performance data
  • Work across multiple teams and organizations to support the design and delivery of best in class on-device ML hardware and software stack
  • Generate and present ML performance data to internal and external teams and stakeholders
  • Work across multiple teams and organizations to support the design and delivery of best in class on-device ML software stack

Qualifications

  • Experience with ML inference, quantization, performance and accuracy
  • Familiarity and experience with the most popular ML architectures (e.g. LLM’s, Diffusion models, CNN’s)
  • A passion to explore and learn about the latest advances in ML model design and architecture, particularly as related to model implementation on HW and on-device inference
  • Familiarity with Operating Systems, embedded systems, and CPU/GPU/SoC/Memory HW architectures
  • Highly proficient in Python/C++ and shell scripting
  • Familiarity with Linux or macOS
  • Exceptional clarity in verbal and written communication, including the ability to summarize, present and lead discussions in larger groups
  • Masters or PhDs in Computer Science or relevant disciplines.
  • Experience with Apple’s CoreML, MPS Graph, Metal Performance Shader’s or MLX frameworks
  • Experience with any ML authoring framework (PyTorch, TensorFlow, JAX, etc.)
  • Experience with implementation of high performance compute kernels for CPU, GPU or AI Accelerators
  • Experience with Apple’s App development framework such as Xcode, Swift, Objective-C
  • Experience with any on-device ML stack, such as TFLite, ONNX, ExecuTorch, etc.
  • Experience with any compiler stack (MLIR/LLVM/TVM etc.)

Skills

Required:ML inferenceML architecturesPython/C++Shell scriptingOperating systems

Benefits

  • Comprehensive medical and dental coverage
  • retirement benefits
  • range of discounted products and free services
  • reimbursement for certain educational expenses — including tuition
  • discretionary bonuses or commission payments as well as relocation (role may be eligible)
  • employee stock programs including discretionary RSU awards and Employee Stock Purchase Plan
Last updated Sep 4, 2026. Interested? Apply on the company site.

About Apple

Discover the innovative world of Apple and shop everything iPhone, iPad, Apple Watch, Mac, and Apple TV, plus explore accessories, entertainment, and expert device support.

Discover the innovative world of Apple and shop everything iPhone, iPad, Apple Watch, Mac, and Apple TV, plus explore accessories, entertainment, and expert device support.

AirPods Max 2Apple ArcadeApple CardApple Fitness+Apple MusicApple TVApple TV+Apple Watch Series 11
Company profileWebsite

More open roles at Apple