Enable real-time, low power, and high performance execution of Deep Learning workloads on Apple products such as Apple Vision Pro, iPhone, iPad, Apple Watch, and Mac
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 with a consistent track record of building high quality production software
- 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
Responsibilities
- 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 and mentor engineers
Other
- Bachelor’s degree in Computer Science, Computer Engineering or equivalent and 5+ years experience shipping products
- Comprehensive medical and dental coverage, retirement benefits, a range of discounted products and free services, and for formal education related to advancing your career at Apple, reimbursement for certain educational expenses - including tuition
- Discretionary restricted stock unit awards, and can purchase Apple stock at a discount if voluntarily participating in Apple’s Employee Stock Purchase Plan
- Discretionary bonuses or commission payments as well as relocation
- Apple is an equal opportunity employer that is committed to inclusion and diversity
