Machine Learning Compiler Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

Apple · 2 months ago

Machine Learning Compiler Engineer

Apple is on the cutting edge of delivering transformative experiences through Artificial Intelligence. As a Machine Learning Compiler Engineer on the Apple Neural Engine team, you'll work to bring high-performance, low-power AI solutions to life on iconic Apple products while collaborating with cross-functional teams.

AppsArtificial Intelligence (AI)BroadcastingDigital EntertainmentFoundational AIMedia and EntertainmentMobile DevicesOperating SystemsTVWearables
check
Comp. & Benefits
check
H1B Sponsor Likelynote

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

Qualification

Compiler designC++MLIRLLVMDeep learning frameworksJIT compilationCommunication skillsMentoring

Required

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

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

Benefits

Comprehensive medical and dental coverage
Retirement benefits
A range of discounted products and free services
Reimbursement for certain educational expenses — including tuition
Discretionary bonuses or commission payments
Relocation

Company

Apple is a technology company that designs, manufactures, and markets consumer electronics, personal computers, and software.

H1B Sponsorship

Apple has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (6998)
2024 (3766)
2023 (3939)
2022 (4822)
2021 (4060)
2020 (3656)

Funding

Current Stage
Public Company
Total Funding
$5.67B
Key Investors
Berkshire HathawayMicrosoftSequoia Capital
2025-05-05Post Ipo Debt· $4.5B
2025-01-16Post Ipo Debt· $0.31M
2021-04-30Post Ipo Equity

Leadership Team

leader-logo
Tim Cook
CEO
leader-logo
Craig Federighi
SVP, Software Engineering
Company data provided by crunchbase