Apple is seeking an engineer to join the ML Compute team and work on large-scale pre-training initiatives for cutting-edge foundation models, distributed training techniques, and optimization of workloads.
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: Kubernetes, Ray, PySpark
Ability to clearly and concisely communicate technical and architectural problems
Responsibilities
Drive large-scale pre-training initiatives for foundation models
Enhance distributed training techniques
Research and implement new patterns and technologies
Optimize execution and performance of workloads
Architect a robust MLOps platform
Operationalize large-scale ML workloads on Kubernetes
Lead complex technical projects
Collaborate with cross-functional engineers
Mentor engineers
Cultivate a team centered on collaboration, technical excellence, and innovation
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.
