Pre-Silicon ML & GPU Compute Framework Engineer

Apple

Cupertino (CA)

On-site

USD 190,000 - 270,000

Full time

5 days ago

Be an early applicant

Application generator

An application made for this job — a tailored resume and cover letter that speak straight to the posting.

Get past ATS filters

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.