← All roles

On-Device ML Compiler Engineer, Model Compilation, Graphics, Games and Machine Learning

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

About the role

ML Infrastructure Engineer role focused on model compilation within the On-Device ML team. Develops and enhances an MLIR-based compiler stack to target Apple hardware (neural engine, GPU, CPU) for efficient on-device execution across devices from Apple Watch to Macs with Ultra SoCs. Collaborates with model authoring, runtime, and performance teams to enable end-to-end ML development and optimized deployment on Apple silicon.

Responsibilities

  • Inspire changes in the MLIR-based compiler to improve runtime performance by leveraging hardware capabilities.

Qualifications

  • 3-5 years working on MLIR-based compilers.
  • Familiarity with common ML model architectures, execution schemes, and operations.
  • Familiarity with C++
  • Familiarity with PyTorch or related training frameworks
  • Familiarity with Swift.
  • Familiarity with programming paradigms for the GPU, CPU, and Neural Engine.
  • Familiarity with writing kernels for ML model execution.

Skills

Required:MLIR-based compiler developmentC++PyTorchGPU/CPU/Neural Engine programming
Nice to have:Swift

Benefits

  • Comprehensive medical and dental coverage
  • retirement benefits
  • discounted products and free services
  • education reimbursements (tuition)
  • potential discretionary bonuses or commission
  • relocation eligibility
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