Apple is looking for a staff engineer to work on ML Compute team, focusing on developing infrastructure for running large-scale workloads on the Cloud and optimizing platform efficiency.
Requirements
Bachelors in Computer Science, engineering, or a related field
6+ years of hands-on experience in building scalable backend systems for training and evaluation of machine learning models
Proficient in relevant programming languages like Python or Go
Strong expertise in distributed systems, reliability and scalability, containerization, and cloud platforms
Proficient in cloud computing infrastructure and tools such as Kubernetes, Ray, PySpark
Ability to clearly and concisely communicate technical and architectural problems, while working with partners to iteratively find solutions
Responsibilities
Lead the development of infrastructure for large-scale workloads on the Cloud
Optimize platform efficiency and throughput
Integrate new features from core distributed computing and ML frameworks
Enhance platform's scalability, performance, and observability
Drive architectural evolution of the platform
Automate and streamline operational processes
Mentor engineers in areas of expertise
Benefits
Employees at Apple are often offered comprehensive benefits that support physical and mental well-being—flexible medical plans, confidential counseling, onsite wellness centers at major campuses, and resources for fitness and daily life. Families typically receive fertility support, paid parental leave with gradual return, caregiving leave, and dependent-care guidance, while financial perks commonly include stock grants (with purchase discounts), 401(k) matching, and income-protection coverage. Employees also see robust time off, Apple University learning and tuition reimbursement, donation matching and paid volunteer hours, and deep product and partner discounts.
