Machine Learning Compiler Engineer

Apple Inc.; Sunnyvale (CA); On-site; USD 150,000 - 190,000; Full time

Job summary

Apple Inc. in Sunnyvale, CA is seeking a Machine Learning Compiler Engineer to advance the Neural Engine compiler stack. You will architect and optimize for deep learning inference, collaborating with hardware and platform teams to enable next‑gen features on Apple devices.

Ideal candidates bring strong C++, compiler design expertise, and hands‑on experience with MLIR/LLVM, IR analysis, and robust software architecture across large codebases.

Qualifications

• Bachelor’s degree in Computer Science, Computer Engineering, or related field with 3 years of experience.
• Experience with program analysis and IR, MLIR and LLVM.
• Expertise in compiler design, including optimizations and code generation.
• Strong C++ skills and large software systems experience.

Responsibilities

• Architect and develop the compiler for Apple's Neural Engine Accelerator.
• Collaborate with hardware/platform teams to support new features.
• Lead design and implementation of complex compiler features.
• Define new compiler architectures balancing performance, energy, and hardware constraints.
• Drive AI workload performance improvements on Apple hardware.

Skills

MLIR/LLVM

Compiler design

C++

Program analysis

Register allocation

Education

Bachelor's degree in CS/CE

Tools

MLIR

LLVM

Job description

Sunnyvale, California, United States Machine Learning and AI

At Apple, we're on the cutting edge of delivering transformative experiences through Artificial Intelligence. If you're passionate about pushing the boundaries of AI and hardware optimization, we want you to join our team! As a Machine Learning Compiler Engineer on the Apple Neural Engine (ANE) team, you'll work to bring high-performance, low-power AI solutions to life on iconic Apple products like the Vision Pro, iPhone, iPad, Mac, and more. This is a dynamic opportunity to work with us in a creative, collaborative environment while developing groundbreaking technologies that will shape the future of computing. Are you ready to help us deliver the next groundbreaking Apple products?

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
• 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

Minimum Qualifications

• Bachelor’s degree in Computer Science, Computer Engineering, or a related field with 3 years of relevant experience
• 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

Preferred Qualifications

• Master's or PhD degree in Computer Science, Computer Engineering, or a related field
• Demonstrated ability to ship high‑quality production software
• Strong communication skills and ability to collaborate effectively across teams and functions
• 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

Apple is an equal opportunity employer that is committed to inclusion and diversity. We seek to promote equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics. Learn more about your EEO rights as an applicant