Job Description

As a staff software engineer on the ML Compute team at Apple, you will design, build, and maintain the compute infrastructure for machine learning applications. You will collaborate with various teams to develop high-impact features and lead a group of engineers while mentoring junior staff.

Requirements

Bachelors in Computer Science
10+ years of industry related experience, working in collaborative environments
Experience with public cloud infrastructure like: Kubernetes, Amazon EC2 and EKS, Google Cloud Platform
Strong software development skills, with proficiency in relevant languages (ex. Golang, Python)
Strong problem solving skills and ability to write performant and high-quality code
Proficient understanding of the software development process, including unit testing and release management
Excellent interpersonal skills; able to work independently as well as in a team; can take feedback and iterate on a solution in a collaborative setting

Responsibilities

Own the architecture, design, development, and operations of large-scale systems designed for machine learning.
Develop custom scheduling, resource management solutions, and fleet management for our ML model training compute infrastructure.
Collaborate with multi-functional teams, integrate with Kubernetes in on-premises and cloud provider clusters, and enable seamless integration with NVIDIA GPUs and other ML accelerators.
Partner with data scientists and machine learning engineers across different Apple organizations to define high-impact product features and deliver them with quality.
Lead a group of engineers to deliver high-quality products/services.
Support junior engineers by providing advice, mentoring, and educational opportunities.

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.