ANE Pre-Silicon Driver Engineer

Apple; Sunnyvale (CA); On-site; USD 180,000 - 260,000; Full time

Job summary

Apple ANE Software team is building the compiler and runtime for machine learning accelerators on Apple Silicon. The role focuses on hardware-accelerated AI execution with real-time, low-power performance across Apple platforms.

You will analyze software interfaces to hardware, develop across framework and firmware, and collaborate with model teams to optimize ML workloads. Strong C/C++/Python and compiler experience are required, with preferences for MS/PhD in CS/ML and MLIR knowledge.

Qualifications

• BS degree required in computer science or related field.
• Experience with compiler development and performance profiling.
• Strong programming skills in C, C++ or Python.

Responsibilities

• Analyze and implement software interfacing with the hardware for AI applications across Apple product platforms.
• Develop software across framework, driver and firmware.
• Collaborate with model teams to support new ML models on Apple Silicon hardware.
• Work cross-functionally with hardware and software teams to resolve issues.
• Conduct performance profiling and diagnostics using specialized tools.
• Engage with QA teams.

Skills

C/C++/Python

Compiler development

Performance profiling

SoC low level software development

Data science basics

Swift

Education

BS in Computer Science

MS or PhD in CS/ML

Tools

MLIR

Swift

Profiling tools

Job description

Summary

The Apple Neural Engine Software team develops the compiler and runtime for machine learning accelerators on Apple Silicon.

Description

In our ANE software team, we are dedicated to providing hardware acceleration using the Apple Neural Engine to enable real time, low power and high performance execution of Deep Learning workloads. Our success is the result of very dynamic people working in an environment which cultivates creativity, partnership and cross-functional collaboration.

Key Responsibilities

• Analyze and implement software interfacing with the hardware for artificial intelligence applications across a wide range of Apple product platforms
• Develop software across our framework, driver and firmware
• Collaborate closely with model teams to support new machine learning models on our hardware
• Work cross functionally with hardware and software teams to resolve issues
• Conduct performance profiling and diagnostics using and developing specialized tools
• Engage with QA teams

Minimum Qualifications

• BS and a minimum of 10 years experience with system software development
• Experience with compiler development
• Excellent programming skills in C, C++ or Python

Preferred Qualifications

• MS or PhD in computer science, machine learning or related field
• Experience in low level software architecture, modeling or validation
• Familiarity with AI networks for example CNN, transformer and diffusion model architectures and their performance characteristics
• Proficiency with profiling and optimizing complex software
• Experience with SoC low level software development, distributed computing or hardware acceleration
• Strong background in data science and statistical methods, with demonstrated ability to analyze large datasets and present complex insights clearly
• Experience with MLIR
• Experience with Swift
• Strong written and verbal communication skills