GPU Engineer, Platform Architecture

$175,800 - $264,200/Yr

Apple - Cupertino, CA

Full-time - Mid Level

Cupertino, CA

Computer and Electronic Product Manufacturing

About the position

The GPU Engineer role at Apple focuses on performance analysis and development for linear algebra and machine learning algorithms on Apple GPU platforms. The position involves collaboration with software engineering and hardware technology teams to address performance issues and enhance AI performance. The successful candidate will also contribute to prototype software implementations and communicate 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 Technologies teams.
• Build prototype software implementations for simulated future hardware and communicate findings to teams building 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.
• 7 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.
• Discretionary bonuses or commission payments.
• Relocation assistance.