GPU ML Engineer

Apple Inc

Confirmed live 2 days ago High trust

Quick summary

Work type
On-site
Location
Cupertino, CA
Salary
$184,700–$324,800 / yr
Posted
23 days ago
Freshness
Confirmed live 2 days ago

Market check

Salary context

Above market

How this pay compares to similar roles

Similar $216k
This role $255k
$144k most similar roles pay here $344k

This role pays more than 88% of similar roles. Most pay $188,562–$244,000 — the shaded band above. At the midpoint, this role pays about $255k versus about $216k for comparable roles.

Based on 240 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 1984 open roles on FindRole.

Listed pay typically runs $175,000–$277,600 across 1590 roles with salary data.

Most-posted roles

View all roles at Apple Inc

At a glance

TL;DR · GPU ML Engineer

The GPU ML Engineer joins the Compute Frameworks team within the GPU, Graphics and Displays organization to develop high-performance data parallel algorithms for iOS, macOS, and Apple TV. This role focuses on providing robust compute solutions to accelerate machine learning networks on Apple Silicon. Key responsibilities include adding optimized GPU compute kernels across machine learning, image processing, linear algebra, and computer vision fields while defining and implementing APIs in Metal Performance Shaders. The engineer will perform compiler and kernel level optimizations, tune GPU-accelerated compute across products, and partner with Platform Architecture teams to influence the hardware roadmap for next generation architectures. Required skills include expertise in GPU compute kernel framework development, system level programming, computer architecture, and high performance parallel programming. Candidates should possess a strong understanding of machine learning fundamentals and numerical methods.

What you'll do

  • Develop optimized GPU compute kernels for machine learning, image processing, and linear algebra.
  • Define and implement APIs within the Metal Performance Shaders framework.
  • Perform compiler and kernel-level optimizations to maximize performance across various hardware families.
  • Tune GPU-accelerated compute performance across multiple product lines.
  • Collaborate with architecture teams to define the roadmap for future Apple GPU compute hardware.
  • Analyze and optimize performance on upcoming silicon architectures.

What we're looking for

  • Technical BS/MS degree or equivalent experience.
  • 2 plus years of experience.
  • Experience with GPU compute kernel framework development, maintenance, and optimization.
  • Experience with system level programming and computer architecture.
  • Experience with high performance parallel programming and GPU programming.
  • Excellent programming, problem-solving, communication, and collaboration skills (preferred).
  • Good understanding of machine learning fundamentals (preferred).
  • Background in mathematics, including linear algebra and numerical methods (preferred).

More like this

Similar roles

GPU ML Engineer

Apple Inc

Cupertino, CA 50 days ago $150,400$277,600
Metal Performance Shaders GPU Programming Machine Learning Linear Algebra Computer Vision Parallel Programming Apple Silicon Numerical Methods
2+ yrs exp

GPU Kernel & Firmware Engineer, Graphics, Games, & ML

Apple Inc

Cupertino, CA 17 days ago $184,700$324,800
C C++ GPU Architecture Firmware Memory Management Power Management Machine Learning Graphics iOS macOS Computer System Architecture Control Theory Asynchronous Command Processing