Job Description

Apple's Intelligence System Experience team is seeking a senior machine learning engineer to develop scalable software systems and algorithms for AI-driven experiences. The role focuses on optimizing training and inference performance, particularly using PyTorch, and offers opportunities for open-source contributions.

Requirements

PhD or Master's degree in Computer Science or equivalent experience
3+ years working with AI/ML technologies in production or research settings
Strong Python programming skills
Understanding of software design principles and algorithms
Experience with deep learning frameworks, such as PyTorch
Experience building large-scale distributed systems
Familiarity with parallelization algorithms for large model training
Familiarity with recent developments in foundation model architectures

Responsibilities

Develop core components for a scalable ML platform
Push the limits of existing training technologies
Create new techniques to overcome system constraints
Optimize training and inference performance using PyTorch
Improve distributed training throughput and system-level efficiency

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.