Machine Learning Compiler Engineer

Job Description

Join Apple's Neural Engine software team to develop compiler technology for deep learning applications on Apple products.

Requirements

• 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
• Bachelor’s degree in Computer Science, Computer Engineering or equivalent and 5+ years experience shipping products

Responsibilities

• Architect and develop the compiler for Apple proprietary Neural Engine Accelerator architecture
• 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 and mentor engineers