GPU ML Engineer

Apple Inc

Confirmed live 2 days ago High trust

Quick summary

Work type
On-site
Location
Cupertino, CA
Salary
$150,400–$277,600 / yr
Posted
50 days ago
Freshness
Confirmed live 2 days ago

Market check

Salary context

Competitive pay

How this pay compares to similar roles

Similar $217k
This role $214k
$135k most similar roles pay here $293k

This role pays more than 57% of similar roles. Most pay $188,562–$244,925 — the shaded band above. At the midpoint, this role pays about $214k versus about $217k 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 domains. The engineer will define and implement APIs in Metal Performance Shaders while performing compiler and kernel level optimizations for various hardware families. Additionally, the role involves partnering with Platform Architecture teams to influence next-generation GPU architectures and roadmap development. 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 linear algebra.

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 hardware.
  • Analyze and optimize performance on upcoming silicon architectures.

What we're looking for

  • Technical BS/MS degree or equivalent experience.
  • Experience in 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.
  • Strong understanding of machine learning fundamentals.
  • Background in mathematics, including linear algebra and numerical methods.
  • Excellent programming, problem-solving, communication, and collaboration skills.

More like this

Similar roles

GPU ML Engineer

Apple Inc

Cupertino, CA 23 days ago $184,700$324,800
Metal Performance Shaders GPU Programming Machine Learning Linear Algebra Computer Vision Parallel Programming Computer Architecture Image Processing 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

On-Device ML Compiler Engineer

Apple Inc

Cupertino, CA 58 days ago $150,400$277,600
MLIR C++ PyTorch Swift GPU Neural Engine Compiler Optimization Machine Learning Model Compression System Software Profiling Debugging Benchmarking