Apple

Staff/Sr. AI Infra Performance Engineer

Apple

United States of AmericaPIPEPosted Jan 23, 2026

Job description

Scaling machine learning workloads across thousands of GPUs and TPUs creates challenges that few engineers ever encounter. In Apple’s Machine Learning Platform Technologies organization, we build the infrastructure that powers large-scale ML training and inference workloads, bringing together expertise in distributed systems, machine learning infrastructure, and high-performance computing.

See all open roles at Apple or explore thousands more companies on TheJobsMap.