Apple is looking to optimize AI and hardware for high-performance, low-power AI solutions on their devices, specifically through the development of a Machine Learning Compiler for the Apple Neural Engine (ANE).
Requirements
- Experience with program analysis and IR (Intermediate Representation), or programming language design, particularly with MLIR and LLVM
- Proven expertise in compiler design and architecture, including deep experience with front-end and middle-end optimizations, register allocation, and back-end code generation
- High-level proficiency in C++ and experience working with large, complex software systems
- Experience optimizing compilers for distributed, parallel, or heterogeneous execution environments, with a solid understanding of shared memory, synchronization, and multi-threading techniques
- Expertise in neural network inference on specialized SoCs or GPUs, and knowledge of deep learning frameworks and tools
- Familiarity with Just-in-Time (JIT) compilation and dynamic optimization techniques for real-time code execution
Responsibilities
- Architect and develop the compiler for Apple's proprietary Neural Engine Accelerator, optimizing it for deep learning inference with a focus on performance, scalability, and power efficiency
- Collaborate with cross-functional teams, including hardware and platform architecture teams, to bring new hardware silicon to market and ensure compiler support for next-gen features
- Lead the design and implementation of complex compiler features, advancing both technical capabilities and strategic alignment across the team and company
- Mentor and guide emerging and mid-level engineers, sharing expertise to foster their growth in compiler technologies and ensuring high standards of code quality, innovation, and efficiency
- Play an instrumental role in defining new compiler architecture approaches and optimizations, balancing trade-offs between performance, energy efficiency, and hardware constraints
- Identify and drive initiatives that will improve the scalability and general performance of AI workloads on Apple hardware, contributing to the vision and roadmap of the Apple Neural Engine team
- Cultivate a culture of learning and collaboration, encouraging innovation, and supporting engineers in their development, with the potential to expand influence across the organization
Other
- Bachelor’s degree in Computer Science, Computer Engineering, or a related field with 5 years of relevant experience
- Demonstrated ability to ship high-quality production software
- Strong communication skills and ability to collaborate effectively across teams and functions
- Proven track record in mentoring and coaching engineers, with an interest in taking on increasing responsibilities and contributing to the team's development
