Pre-Silicon ANE Driver Engineer for Real-Time ML

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

Benefits offered by this job

Stock options

Medical & dental coverage

Retirement benefits

Job summary

Apple in Sunnyvale is seeking a software engineer for the ANE Software team to drive hardware-accelerated ML workloads. You will analyze and implement software interfacing with the hardware across Apple product platforms, and contribute to the framework, driver and firmware.

Collaborate with model teams to support new ML models on our hardware, and work cross-functionally with hardware and software groups to resolve issues.

Qualifications

• BS and a minimum of 3 years experience with system software development.
• Experience with compiler development.
• Excellent 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 our framework, driver and firmware.
• Collaborate closely with model teams to support new ML 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.

Skills

C/C++/Python

Profiling

Parallel computing

Education

BS in Computer Science

MS/PhD in CS or ML

Tools

MLIR

Swift

Job description

Apple in Sunnyvale is seeking a software engineer for the ANE Software team to drive hardware-accelerated ML workloads. You will analyze and implement software interfacing with the hardware across Apple product platforms, and contribute to the framework, driver and firmware.

Collaborate with model teams to support new ML models on our hardware, and work cross-functionally with hardware and software groups to resolve issues.