← All roles

GPU ML Engineer

Apple· Software and Services
Apply on company site
Cupertino, CA$150.4k - $277.6kFull-timePosted 3mo ago

About the role

Apple’s Compute Frameworks team develops high-performance data parallel algorithms for ML, image processing, linear algebra, and computer vision across iOS, macOS, and Apple TV. The role focuses on accelerating machine learning networks on Apple Silicon, with influence on compute and programming models in next-gen GPU architectures.

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

  • Technical BS/MS degree and Equivalent experience.
  • 2 plus years of experience
  • GPU compute kernel framework development, maintenance, and optimization.
  • Experience with system level programming and computer architecture.
  • Experience with high performance parallel programming, GPU programming experience
  • Excellent programming and problem-solving skills.
  • Strong communication skills.
  • Strong collaboration skills.
  • Good understanding of machine learning fundamentals.
  • Background in mathematics, including linear algebra and numerical methods is a plus.

Skills

Required:GPU compute kernel developmentMetal Performance Shaderssystem level programminghigh performance parallel programmingmachine learning fundamentals

Benefits

  • Comprehensive medical and dental coverage
  • retirement benefits
  • a range of discounted products and free services
  • reimbursement for certain educational expenses — including tuition
  • discretionary bonuses or commission payments as well as relocation (role might be eligible)
Last updated Jul 24, 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