← All roles

On-device ML Infrastructure Engineer (Orchestration & Performance)

Apple· Machine Learning and AI
Apply on company site
Cupertino, California$150.4k - $277.6kFull-timePosted 1mo ago

About the role

ML Infrastructure Engineer focusing on model orchestration and performance for on-device execution across Apple Silicon, collaborating with authoring, compiler, and runtime teams to enable stable and high-performance model execution on Apple devices.

Responsibilities

  • Drive full-stack changes through the OS and tooling to enable deploying large SOTA models across the Apple Silicon ecosystem, from Apple Watch to Mac Studios.
  • Partner with teams across the company to support advanced use cases for Siri, Apple Intelligence, Camera, and more.
  • Make changes in our authoring and MLIR-based compiler to expose mechanisms for state-of-the-art model execution
  • Implement mechanisms to support efficient orchestration of models across Apple Silicon hardware.

Qualifications

  • 3-5 years working on tooling built in Python 3 and C++/Swift
  • Familiarity with common ML model architectures, execution schemes, and operations.
  • Familiarity with PyTorch or related training frameworks.
  • Experience working on or adjacent to MLIR-based compilers.
  • Familiarity with deploying applications or tooling on Apple platforms.
  • Familiarity with programming paradigms for the GPU, CPU, and Neural Engine.
  • Familiarity with writing kernels for ML model execution.

Skills

Required:PythonC++/SwiftML model architectures
Nice to have:PyTorchMLIRApple platformsGPU/CPU/Neural Engine paradigmsKernel development

Benefits

  • Comprehensive medical and dental coverage
  • retirement benefits
  • discounted products and free services
  • reimbursement for certain educational expenses — including tuition
  • discretionary bonuses or commission payments
  • relocation eligibility
  • Apple shareholder opportunities through stock programs
  • eligibility for Employee Stock Purchase Plan
Last updated Oct 16, 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