GPU Compute Performance Engineer
Description For GPU Compute Performance Engineer
Apple Silicon GPU Driver Performance team is seeking an exceptional GPU Compute Performance Engineer to join their elite team. This role sits at the intersection of GPU architecture, machine learning, and performance optimization, working on cutting-edge technology that powers Apple's entire product line.
The position offers a unique opportunity to work with Apple's GPU hardware architecture and design teams, helping shape the future of Apple Silicon GPUs. You'll be responsible for optimizing machine learning workloads and GPU compute applications, ensuring peak performance across Apple's platforms. The role involves working with various internal teams including Apple Intelligence, Private Cloud Compute, and MetalFX.
As a GPU Compute Performance Engineer, you'll be deeply involved in all phases of product development - from architectural planning to performance analysis and optimization. You'll work with both internal and external partners to achieve maximum GPU performance, developing and utilizing state-of-the-art analysis tools and capabilities.
The ideal candidate combines deep technical expertise in GPU architecture and programming models with a passion for performance optimization. You'll have the opportunity to work on diverse projects including benchmarks, ML training and inference, and various GPU compute applications. This role offers the chance to make a significant impact on Apple's future products while working with some of the industry's best hardware and software engineers.
Join Apple's GPU Driver Performance team to be at the forefront of GPU compute technology, working on challenging problems that affect millions of users worldwide. This role offers excellent growth opportunities and the chance to contribute to Apple's continued innovation in GPU technology and machine learning acceleration.
Responsibilities For GPU Compute Performance Engineer
- Work with internal partners to analyze and improve GPU and system performance of large scale ML deployments
- Optimize GPU based ML algorithm implementations and compute applications for best performance
- Work with hardware teams to define hardware roadmap for GPU performance
- Develop tools and frameworks to support performance analysis on Apple Silicon GPUs
Requirements For GPU Compute Performance Engineer
- Experience or interest in emerging GPGPU use cases in ML and compute
- Experience or interest in optimizing compute workloads for GPU performance
- GPU programming with Metal, DirectX, Vulkan, CUDA, Direct Compute, OpenGL, or OpenCL
- Excellent software design and problem solving skills
- Excellent system debugging skills
- Excellent written and oral communication skills
- Experience in ML frameworks such as pytorch, tensorflow, JAX (preferred)
- Experience in GPU compute kernel optimization for ML training and inference operations (preferred)
Jobs Related To Apple GPU Compute Performance Engineer
