ML Framework (MetalLM) Engineer at Apple

Department: Software and Services

Compensation · Listed in posting

$150,400 – $225,300

Location: Cupertino, California
Level: mid
Posted: Aug 24, 2026

Description

Apple is seeking a Machine Learning and GPU Programming Engineer to develop and optimize scalable, production-grade ML inference and training solutions on Apple Silicon and custom server hardware. The role focuses on distributed compute strategies, kernel and compiler optimizations, model optimization techniques, performance analysis, and Metal device-backend development, while collaborating with hardware, compiler, and systems teams.

Required Skills

C/C++/ObjC
GPU kernels
Distributed compute
Metal backend
Quantization
Computer architecture

Additional Information

Required Previous Experiences: 2+ years of programming and problem-solving experience with C/C++/ObjC
Experience with GPU kernel development and optimizations using Metal or CUDA
Experience with system-level programming and computer architecture
Experience with distributed training or inference techniques
Preferred Qualifications: Experience with graph compilers such as Triton, OpenXLA, or LLVM/MLIR
Contributions to an AI framework such as PyTorch, JAX, or TensorFlow
Understanding of machine learning fundamentals

About Apple

Apple

Mac laptops and desktopsCupertino, CA

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.

Customers and case studies

See customers and case studies