GPU ML Engineer
Core
Developing high-performance data parallel algorithms and compute kernels for accelerating machine learning networks on Apple Silicon.
Role type
Senior IC GPU ML Engineer
Builds
Optimized GPU compute kernels, Metal Performance Shaders APIs, and compute solutions for iOS, macOS, and Apple TV.
Domain
Consumer electronics, GPU architecture, machine learning
Deliverable
production ML models
Required skills
GPU compute kernel development, Metal Performance Shaders, system-level programming, computer architecture, high-performance parallel programming, compiler optimization, linear algebra, image processing
Preferred skills
Machine learning fundamentals, mathematics (linear algebra, numerical methods)
Technologies
Metal Performance Shaders, Apple Silicon
Responsibilities
Adding optimized GPU compute kernels for ML, image processing, linear algebra, and computer vision; Defining and implementing APIs in Metal Performance Shaders; Performing compiler and kernel-level optimizations; Partnering with Platform Architecture teams to define GPU hardware roadmap; Tuning GPU-accelerated compute across products.
Seniority
Mid-level, hands-on IC
