Apple

Machine Learning Compiler Engineer

Apple · Sunnyvale, California, United States

Computers and Electronics Manufacturing · 10,001+ employees

8 h ago
Mid (2-5 yrs) Full-time United States
Log in to apply, save this posting, or score it against your profile with AI.

About the role

You will develop high-performance, low-power AI solutions for Apple products by working on the Apple Neural Engine compiler. This role involves collaborating in a creative environment to build groundbreaking technologies that shape the future of computing.

What they look for

Machine Learning Compiler Design MLIR LLVM C++ Program Analysis Intermediate Representation Register Allocation Code Generation Neural Network Inference SoC Optimization GPU Optimization Deep Learning Frameworks JIT Compilation Parallel Computing Multi-threading

Requirements

Candidates must have a bachelor's degree in a relevant field and at least 3 years of experience in compiler design, program analysis, and IR. Proficiency in C++ and experience with MLIR or LLVM are essential for this position.

Full description

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!

Description

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?

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