GPU ML Engineer - High-Performance ML on Silicon

Apple Inc.; Cupertino (CA); On-site; USD 150,400 - 277,600; Full time

Benefits offered by this job

Medical & dental

Retirement benefits

Employee stock programs

Tuition reimbursement

Relocation assistance

Job summary

Apple Inc. in Cupertino, CA, is seeking extraordinary machine learning and GPU programming engineers to deliver robust compute solutions on Apple Silicon. You will contributes to optimized kernels, ML workflows, and high-performance GPU architectures across iOS, macOS and Apple TV.

The role emphasizes collaboration with architecture teams, API work in Metal, and kernel-level tuning to maximize performance across Apple devices. A strong foundation in parallel GPU programming is essential.

Qualifications

• BS/MS degree or equivalent experience required.
• 2+ years of hands-on experience in GPU compute kernel development and optimization.
• Experience with high performance parallel programming and GPU programming across compute platforms.

Responsibilities

• Implement optimized GPU kernels for ML, image processing, linear algebra, and computer vision tasks.
• Define and implement APIs in Metal Performance Shaders.
• Conduct in-depth performance analysis and kernel-level optimizations for Apple hardware.
• Collaborate with Platform Architecture to influence compute hardware roadmaps for Apple Silicon.

Skills

GPU compute kernel

Parallel programming

System programming

Computer architecture

GPU programming

Education

BS/MS degree or equivalent

Job description

Apple Inc. in Cupertino, CA, is seeking extraordinary machine learning and GPU programming engineers to deliver robust compute solutions on Apple Silicon. You will contributes to optimized kernels, ML workflows, and high-performance GPU architectures across iOS, macOS and Apple TV.

The role emphasizes collaboration with architecture teams, API work in Metal, and kernel-level tuning to maximize performance across Apple devices. A strong foundation in parallel GPU programming is essential.