Pre-silicon Compute and ML Framework Manager

Apple; Cupertino (CA); On-site; USD 180,000 - 240,000; Full time

Job summary

Apple seeks an engineering manager for the Pre-silicon Compute Frameworks team to lead API development and performance analysis for Metal and GPU compute features in future Apple Silicon GPUs.

You will collaborate with Apple software and hardware leaders to propose and evaluate hardware features from emulation to silicon, guiding a team of engineers toward a strong software/hardware roadmap.

Qualifications

• BS/MS/PhD in Computer Science or related field.
• 2+ years in a management or lead role.
• Strong grasp of at least one GPU compute API (Metal/CUDA/OpenCL).
• Good understanding of ML fundamentals and familiarity with ML frameworks (PyTorch, JAX, CoreML).
• Knowledge of OS, computer architecture, and CPU/GPU programming.
• Strong background in C/C++ programming.

Responsibilities

• Lead, support, and grow the Pre-silicon Compute Frameworks team to deliver software and APIs for new hardware features.
• Provide direction to ensure APIs enable developers to leverage new hardware features simply.
• Partner with cross-functional teams to align software/hardware roadmaps with state-of-the-art capabilities.
• Contribute to bring-up of new chips and products with new GPUs.

Skills

Leadership

Cross-functional collaboration

ML fundamentals

CPU/GPU programming

Education

BS/MS/PhD in Computer Science or related field

Tools

Metal API

CUDA

OpenCL

PyTorch

JAX

CoreML

Job description

Summary

Apple's Graphics, Games, and Machine Learning team provides the compute and graphics software foundation across all of our innovative products including iPhone, iPad, Apple TV, Mac, Vision Pro, and Apple Watch. The Pre-silicon Compute Frameworks team is responsible for API design and development across Metal and Metal compute frameworks and for functional support and performance analysis of upcoming hardware features in future Apple Silicon GPUs. We are looking for an engineering manager with a strong technical background who is excited to lead engineers and collaborate with other leaders to ensure a strong software and hardware roadmap for our products.

Description

As the manager of the Pre-silicon Compute Frameworks team, you will support, lead, and grow a team of engineers through API development and performance analysis across Metal and Metal compute frameworks for future Apple Silicon GPUs. You will partner with other leaders across Apple software and hardware organizations to propose, develop, and evaluate hardware features for new silicon generations from emulation through silicon.

Key Responsibilities

Lead, support, and grow the Pre-silicon Compute Framework team to deliver software, representing Apple’s roadmap for compute and ML technologies during hardware developmentProvide direction and guidance to ensure the team designs APIs that enable developers to make full use of new hardware features in the simplest way possiblePartner with cross-functional teams to ensure our software and hardware roadmap includes the state-of-the-art features and performanceContribute to the bring-up of new chips and products using our new GPUs

Minimum Qualifications

BS/MS/PhD or equivalent in Computer Science or related field2 - 3 years as a manager or leadStrong grasp and practical experience with at least one GPU compute APIs (Metal, CUDA, OpenCL, etc.)Good understanding of machine learning fundamentals and familiarity with ML frameworks (PyTorch, JAX, CoreML)Knowledge of Operating Systems, Computer Architecture, and CPU/GPU programmingStrong background in C/C++ programming

Preferred Qualifications

Experience defining, planning and executing complex projects involving partnership with multiple teamsProven track record in strategic leadership and strong decision makingFamiliarity with Apple framework and API design patternsHands on experience with ML frameworks (PyTorch, JAX, CoreML)