Your browser does not support javascript! Please enable it, otherwise web will not work for you.

On-Device ML Compiler Engineer

Home > C/C++ programming jobs

On-Device ML Compiler Engineer in USA

  • Cupertino, CA

Responsibilities

  • Develop and improve MLIR-based compiler infrastructure for machine learning model compilation.
  • Target the Neural Engine, GPU, and CPU to maximize hardware capabilities for ML execution.
  • Support model optimization, transformation, execution, debugging, profiling, and analysis across Apple devices.
  • Collaborate with model authoring, runtime, and performance teams.
  • Write kernels and compiler optimizations for efficient ML model execution.

Requirements

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

Apple

We’re a diverse collective of thinkers and doers, continually reimagining what’s possible to help us all do what we love in new ways. And the same innovation that goes into our products also applies to our practices — strengthening our commitment to leave the world better than we found it. This i...

Similar positions

Senior Software Engineer, Simulation

  • ATOMS
  • Full time
  • USA
  • 09/13/2026
  • Salary: $195k - $246k/yr
  • San Francisco, CA

Firmware Engineer II

  • Varda Space Industries
  • Full time
  • USA
  • 09/13/2026
  • Salary: $144k - $184k/yr
  • El Segundo, CA

Senior Software Engineer, Simulation

  • CloudKitchens
  • Full time
  • USA
  • 09/13/2026
  • Salary: $195k - $246k/yr
  • San Francisco, CA

Software Engineer, Battlespace Awareness

  • Anduril Industries
  • Full time
  • USA
  • 09/13/2026
  • Salary: $191k - $253k/yr
  • Waltham, MA

Software Engineer, Battlespace Awareness

  • Anduril Industries
  • Full time
  • USA
  • 09/13/2026
  • Salary: $165k - $218k/yr
  • Broomfield, CO