← All roles

Server ML Frameworks Engineer

Apple· Software and Services
Apply on company site
Cupertino, CA$184.7k - $324.8kFull-timePosted 3d ago

About the role

Senior technical leader on Apple's Server ML Frameworks team responsible for setting technical direction and architecture for scalable, production-grade server ML frameworks optimized for high-throughput GPU execution, in collaboration with software and hardware teams to define roadmaps for Apple Intelligence in Private Cloud Compute.

Responsibilities

  • Adding optimized GPU compute kernels across Machine Learning, Image Processing, Linear Algebra and Computer Vision.
  • Defining and implementing APIs in Metal Performance Shaders.
  • Performing in-depth analysis, compiler and kernel level optimizations to ensure the best possible performance across hardware families.
  • Partnering with Platform Architecture teams to define Apple GPU's compute hardware roadmap. Working with hardware team to analyze performance on future silicon
  • Tune GPU-accelerated compute across products.

Qualifications

  • 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
  • Familiarity with Apple framework and API design patterns
  • Experience with graph compilers such as CuTE, CuTile, Triton, OpenXLA or LLVM is a plus
  • Good understanding of LLM and Diffusion based model architectures

Skills

Required:GPU kernel developmentMetal Performance Shaders APIsC/C++ system programmingDistributed training/inference
Nice to have:GPU hardware/architecture collaboration

Benefits

  • Comprehensive medical and dental coverage
  • retirement benefits
  • discounted products and free services
  • reimbursement for certain educational expenses including tuition
  • discretionary bonuses or commission payments possibly
  • relocation assistance
  • participation in employee stock programs (RSU, ESOP)
Last updated Oct 23, 2026. Interested? Apply on the company site.

About Apple

Discover the innovative world of Apple and shop everything iPhone, iPad, Apple Watch, Mac, and Apple TV, plus explore accessories, entertainment, and expert device support.

Discover the innovative world of Apple and shop everything iPhone, iPad, Apple Watch, Mac, and Apple TV, plus explore accessories, entertainment, and expert device support.

AirPods Max 2Apple ArcadeApple CardApple Fitness+Apple MusicApple TVApple TV+Apple Watch Series 11
Company profileWebsite

More open roles at Apple