Logo
Apple

Machine Learning Compiler Engineer

Apple, Sunnyvale, California, United States, 94087

Save Job

Overview Machine Learning Compiler Engineer

Sunnyvale, California, United States

Machine Learning and AI

Description As a Machine Learning Compiler Engineer on the Apple Neural Engine (ANE) team, you will 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. The role evolves to include leadership, collaboration across engineers, and guiding the direction of Apple’s AI-driven capabilities across the ecosystem.

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

Minimum Qualifications

Bachelor’s degree in Computer Science, Computer Engineering, or a related field with 5 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

Bachelor’s (and/or Master’s/PhD) degree in Computer Science, Computer Engineering, or a related field with 10 years of relevant experience

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

Proven track record in mentoring and coaching engineers, with an interest in taking on increasing responsibilities and contributing to the team's development

Pay & Benefits At Apple, base pay is one part of our total compensation package and is determined within a range. The base pay range for this role is between $181,100 and $318,400, and your base pay will depend on your skills, qualifications, experience, and location. Apple employees also have the opportunity to become an Apple shareholder through participation in Apple’s discretionary employee stock programs and may receive discretionary restricted stock unit awards. Benefits include comprehensive medical and dental coverage, retirement benefits, access to discounted products and free services, reimbursement for certain educational expenses, and potential discretionary bonuses or relocation support. This role might be eligible for education-related reimbursements and other benefits. Learn more about Apple Benefits.

Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.

Equal Opportunity and EEO Rights 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.

Apple will not discriminate or retaliate against applicants who inquire about, disclose, or discuss their compensation. Apple participates in the E-Verify program where required by law. Apple is committed to working with and providing reasonable accommodation to applicants with physical and mental disabilities. Reasonable Accommodation and Drug Free Workplace policy apply. Apple will consider for employment all qualified applicants with criminal histories in a manner consistent with applicable law. If you’re applying for a position in San Francisco, review the San Francisco Fair Chance Ordinance guidelines applicable in your area.

It is unlawful in Massachusetts to require or administer a lie detector test as a condition of employment. An employer who violates this law shall be subject to criminal penalties and civil liability.

#J-18808-Ljbffr