Server ML Frameworks Engineer

What you'll do

• Develop optimized GPU compute kernels and define and implement APIs in Metal Performance Shaders.
• Analyze and optimize compiler- and kernel-level performance across hardware; tune GPU-accelerated compute across products.
• Partner with platform architecture and hardware teams on GPU roadmaps and future silicon; mentor engineers and lead cross-functional initiatives.

What they ask for

• BS/MS/PhD in Computer Science or a related field, or equivalent practical experience.
• 3–5 years of experience as a manager or technical leader.
• Experience developing and optimizing GPU kernels using Metal, CUDA, or similar compute programming models; experience with distributed training or inference techniques.
• Strong system-level programming skills in C/C++, with understanding of operating systems and computer architecture.

Nice to have

• Track record in strategic leadership and decision-making; familiarity with Apple framework and API design patterns.
• Experience with graph compilers such as CuTE, CuTile, Triton, OpenXLA or LLVM; understanding of LLM and Diffusion based model architectures.

Pay

$184,700 and $324,800 USD base salary

Tools

Metal Performance Shaders, Metal, CUDA, C/C++, CuTE, CuTile, Triton, OpenXLA

AI focus

Enables high-performance, distributed inference of generative AI applications, including LLMs, for Apple Intelligence on Private Cloud Compute; supports evaluation of new Apple Intelligence models.