Machine Learning Compiler Engineer

Apple

| Sunnyvale, California, United States (On Site) | Full Time | 5 months ago

Apply Now

Job Summary

As a Machine Learning Compiler Engineer, you will be responsible for architecting and developing the compiler for Apple's proprietary Neural Engine Accelerator. This involves optimizing for deep learning inference, focusing on performance, scalability, and power efficiency. You will collaborate with cross-functional teams to support new hardware silicon and ensure compiler support for next-generation features. Leading the design and implementation of complex compiler features is crucial, as is mentoring engineers and defining new compiler architecture approaches. You will also identify initiatives to improve AI workload performance, contributing to the vision of the Apple Neural Engine team. Cultivating a culture of learning and collaboration is also part of the job responsibilities.

Must Have

Architect and develop the compiler.
Optimize for deep learning inference.
Collaborate with cross-functional teams.
Lead design and implementation of compiler features.
Mentor and guide engineers.

Job Description

As a Machine Learning Compiler Engineer, you will: • 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

3 Skills Required For This Role

Cross Functional Deep Learning Machine Learning

Similar Jobs

Research Development

Senior Machine Learning Engineer – Ads Signals Intelligence & Information Retrieval