Server ML Frameworks Engineer

What you'll do

• Add optimized GPU compute kernels for Machine Learning, Image Processing, Linear Algebra, and Computer Vision.
• Define and implement APIs within Metal Performance Shaders.
• Perform compiler and kernel level optimizations to maximize performance across hardware families.
• Partner with Platform Architecture teams to define the Apple GPU compute hardware roadmap.
• Analyze performance on future silicon to influence next-generation hardware design.
• Tune GPU-accelerated compute performance across various products.
• Set the technical direction and software architecture for frameworks powering Apple Intelligence.

What we're looking 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 with GPU kernel development and optimization using compute programming models such as Metal, CUDA, or similar.
• Experience with distributed training or inference techniques.
• Strong system-level programming skills in C/C++ with a solid understanding of operating systems and computer architecture.
• Track record in strategic leadership and decision-making (preferred).
• Familiarity with Apple framework and API design patterns (preferred).
• Experience with graph compilers such as CuTE, CuTile, Triton, OpenXLA or LLVM (preferred).