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

Embedded Development Engineer

  • Emerson Electric Co.
  • Full time
  • Others
  • 10/04/2026
  • Pune, India

Intermediate Embedded Software Engineer

  • MITRE
  • Full time
  • USA
  • 10/04/2026
  • Salary: $104k - $155k/yr
  • New Bedford, MA

Senior Embedded Software Engineer

  • Apex Technology, Inc.
  • Full time
  • USA
  • 10/04/2026
  • Salary: $185k - $210k/yr
  • Los Angeles, CA

Bluetooth Development Engineer

  • Visteon Corporation
  • Full time
  • Others
  • 10/04/2026
  • Chennai, India

Software Engineer II

  • Microsoft
  • Full time
  • USA
  • 10/04/2026
  • Salary: $102k - $219k/yr
  • Redmond, WA