Pre-Silicon ML and Compute Framework Engineer, Graphics, Games and ML

Apple; Cupertino (CA); On-site; USD 190,000 - 270,000; Full time

Job summary

Apple's Graphics, Games, and Machine Learning team in Cupertino seeks a senior engineer to influence and lead new features in Metal and Metal compute frameworks through our next‑generation GPU hardware roadmap. You will design APIs, implement compute and ML use cases, and partner with cross‑functional teams to ensure software and hardware roadmaps reflect state‑of‑the‑art features and performance.

Strong C/C++ and GPU programming skills are required, with experience in ML frameworks; advanced

Qualifications

• BS/MS/PhD or equivalent in CS or related field with 5 years experience.
• Strong grasp of GPU compute APIs (Metal, CUDA, OpenCL) and software development.
• Familiarity with ML frameworks (PyTorch, JAX, CoreML) and ML fundamentals.
• Knowledge of OS, computer architecture, and CPU/GPU programming.
• Strong C/C++ programming background.

Responsibilities

• Develop and deliver software representing Apple's roadmap for compute and ML technologies during hardware development.
• Design APIs enabling developers to leverage new hardware features simply.
• Partner with cross-functional teams to align software and hardware roadmaps with state-of-the-art features and performance.
• Contribute to bring-up and verification of new chips and products using our GPUs.

Skills

GPU compute APIs

CPP programming

ML fundamentals

OS concepts

Education

BS/MS/PhD in CS or related field

Tools

Metal API

CUDA

OpenCL

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 seeking a senior engineer to influence and lead new features in Metal and Metal compute frameworks through our next-generation GPU hardware roadmap.

Description

The Pre-silicon Compute Frameworks team is responsible for API design and development across Metal and Metal compute frameworks for upcoming hardware features in future Apple Silicon GPUs. As a member of this team, you will contribute to API design, development, and performance analysis across Metal and Metal compute frameworks for future Apple Silicon GPUs. Along with API development, you will develop critical compute and ML use cases to showcase new features. 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.

Responsibilities

• Develop and deliver software representing Apple's roadmap for compute and ML technologies during hardware development.
• Design APIs that enable developers to make full use of new hardware features in the simplest way possible.
• Partner with cross-functional teams to ensure our software and hardware roadmap includes the state-of-the-art features and performance.
• Contribute to the bring-up and verification of new chips and products using our new GPUs.

Minimum Qualifications

• BS/MS/PhD or equivalent in Computer Science or related field with 5 years experience.
• Strong 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 programming
• Strong background in C/C++ programming

Preferred Qualifications

• Experience collaborating in large cross functional projects
• Familiarity with Apple framework and API design patterns
• Hands on experience with ML frameworks (PyTorch, JAX, CoreML)