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 ML compiler infrastructure focused on model compilation and runtime execution.
  • Use the MLIR-based compiler stack to target Apple’s Neural Engine, GPU, and CPU.
  • Collaborate with model authoring, runtime, and performance teams to maximize hardware utilization.
  • Support model optimization, transformation, execution, debugging, profiling, and analysis across Apple devices.
  • Write or improve kernels for machine learning 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 GPU, CPU, and Neural Engine environments.
  • Preferred: familiarity with writing kernels for machine learning 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