Pre-Silicon ML & GPU Compute Framework Engineer

Apple, Inc.; 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

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