Apple · Sunnyvale

Machine Learning Compiler Engineer

(f/m/d) · 1/17/2025

Description

We develop compiler technology to accelerate deep learning applications for Apple products. In this role, you will be empowered to: - Architect and develop the compiler for Apple proprietary Neural Engine Accelerator architecture, to enable inference of deep learning networks onto this architecture with an emphasis on performance and power - Bring up new hardware silicon and add support in the compiler for these hardware features - Work on bringing the compiler code to production quality and enable a wide range of applications of deep learning technology for internal clients and 3rd party developers - Evaluate existing hardware blocks and work closely with the platform architecture team on the definition of new hardware features, and hardware specification review - Work with the micro-architecture design team to understand the functional and performance goals of the design - Architect and lead complex compiler features

Qualifications

  • In-depth working knowledge of compiler architecture, front-end and middle-end optimizations, scheduling, register allocation, back-end code generation
  • Experience with IR and/or programming language design, esp. experience with MLIR and LLVM
  • High level C++ programming expertise and experience working in a large and complex software stack with a consistent track record of building high quality production software
  • Bachelor’s degree in Computer Science, Computer Engineering or equivalent and 10+ years relevant experience

Preferred Qualifications

  • Experience with compiling for distributed and parallel execution environment including distributed execution, shared memory, synchronization, etc.
  • Experience with neural networks inference on dedicated SoC or GPU is highly desired
  • Understanding of JIT techniques for dynamic optimization and execution of code
  • Ability to analyze and optimize program execution paths

Benefits

Application

View listing at origin and apply!