Pre-silicon ML and Compute Framework Engineer, Graphics, Game and ML

Apple Inc

Quick summary

Work type
On-site
Location
Cupertino, CA
Salary
$181,100–$318,400 / yr
Posted
56 days ago

Market check

Salary context

Above market

How this pay compares to similar roles

Similar $211k
This role $250k
$143k most similar roles pay here $337k

This role pays more than 87% of similar roles. Most pay $184,150–$236,900 — the shaded band above. At the midpoint, this role pays about $250k versus about $211k for comparable roles.

Based on 239 similar postings.

Employer

About Apple Inc

Apple Inc. is a multinational technology company known for designing and manufacturing consumer electronics, software, and online services, including the iPhone, Mac, iPad, and App Store. Industry: Consumer Electronics & Software

Apple Inc currently has 1723 open roles on FindRole.

Listed pay typically runs $162,500–$272,100 across 1398 roles with salary data.

Most-posted roles

View all roles at Apple Inc

At a glance

TL;DR · Pre-silicon ML and Compute Framework Engineer, Graphics, Game and ML

Apple’s Graphics, Games, and Machine Learning team is seeking a senior engineer to join the Pre-silicon Compute Frameworks team, responsible for designing and developing APIs for Metal and Metal compute frameworks across future Apple Silicon GPUs. This role involves contributing to API design, performance analysis, and the development of critical compute and ML use cases to showcase new hardware features. You will work closely with cross-functional teams to ensure that software and hardware roadmaps include cutting-edge technologies and collaborate on the bring-up and verification of new chips and products using advanced GPUs. The ideal candidate has a strong background in C/C++ programming, experience with GPU compute APIs like Metal or CUDA, and familiarity with ML frameworks such as PyTorch and CoreML, along with knowledge of operating systems and computer architecture.

What you'll do

  • Develop software representing Apple’s roadmap for compute and ML technologies during hardware development.
  • Design APIs to enable developers to utilize new hardware features efficiently.
  • Partner with cross-functional teams to ensure state-of-the-art features in the software and hardware roadmap.
  • Contribute to the bring-up and verification of new chips and products using future GPUs.
  • Develop critical compute and ML use cases to showcase new GPU hardware features.

What we're looking for

  • BS/MS/PhD or equivalent in Computer Science with 5+ years of relevant experience.
  • Proficient in at least one GPU compute API such as Metal, CUDA, or OpenCL.
  • Solid understanding of machine learning fundamentals and practical experience with ML frameworks like PyTorch, JAX, or CoreML.
  • Expertise in Operating Systems, Computer Architecture, and CPU/GPU programming.
  • Strong C/C++ programming skills.

More like this

Similar roles

ML Framework (MetalLM) Engineer, Graphics, Game and ML

Apple Inc

Cupertino, CA 56 days ago $147,400$272,100
C/C++ CUDA Metal Distributed training Graph compilers LLVM Triton OpenXLA CuTE CuTile System level programming Computer architecture Performance analysis Latency optimization Memory footprint analysis Compute efficiency TensorFlow PyTorch CI/CD

GPU Compiler Engineer, Graphics, Game and ML

Apple Inc

Cupertino, CA 36 days ago $126,800$190,900
Metal CUDA OpenGL OpenCL DirectX PyTorch Python C++ LLVM GCC CI/CD Git Linux macOS GPU Compiler IR TypeSystems MemoryHierarchy ComputerArchitecture

GPU Software Architecture Engineer, Graphics, Games, & ML

Apple Inc

Cupertino, CA 56 days ago $181,100$318,400
CUDA ROCm C/C++ InfiniBand RDMA NCCL PyTorch JAX TensorFlow Distributed Systems Parallel Computing Performance Profiling Pipeline Parallelism Expert Parallelism System Programming ML Infrastructure Python