GPU Architect, Platform Architecture

$207,800 - $312,200/Yr

Apple - Cupertino, CA

Full-time - Senior

Cupertino, CA

Computer and Electronic Product Manufacturing

About the position

The GPU Architect role at Apple involves analyzing and optimizing the performance of linear algebra and machine learning algorithms on Apple GPU platforms. The position requires collaboration with software engineering and hardware technology teams to develop system-level strategies that enhance AI performance. The successful candidate will also be responsible for building prototype software implementations and communicating findings to production software teams.

Responsibilities

• Analyze the performance of linear algebra and machine learning algorithms on Apple GPU platforms.
• Formulate system-level strategies to address performance problems in collaboration with software engineering and hardware technology teams.
• Build prototype software implementations for simulated future hardware.
• Communicate findings and insights to teams developing production software.

Requirements

• BS degree in Computer Science, Electrical Engineering, or equivalent.
• Experience with software and hardware performance analysis and optimization.
• Experience in GPU programming models such as Metal, CUDA, or similar.

Nice-to-haves

• MS or PhD in Computer Science, Electrical Engineering, or equivalent.
• 15 years of relevant industry experience.
• Experience with the internals of GPU drivers, compilers, and/or accelerated libraries.
• Experience working specifically in CUDA C++ on ML and/or linear algebra algorithms.
• Experience with distributed algorithms for HPC (for example, supercomputing).
• Experience communicating across both hardware and software organizations.

Benefits

• Comprehensive medical and dental coverage.
• Retirement benefits.
• Discounted products and free services.
• Reimbursement for certain educational expenses, including tuition.
• Opportunity to participate in Apple's discretionary employee stock programs.