Job Description

Join a team focused on building ML-inference applications and services on Apple Silicon, particularly in generative AI. You will work on improving stability and performance for private cloud compute while integrating new functionalities from research, collaborating with various product teams.

Requirements

Familiarity with Apple ML stack (ANE, CoreML, MPS/Metal)
Experience with high-level general distributed ML stack (PyTorch-distributed, NCCL)
On-device iOS development experience
Quality focus - produce reliable, maintainable, deliverable software
Comfortable diving deep - working across multiple levels of abstraction
Good at handling relationships & communication - collaborate well with colleagues

Responsibilities

Engineer continuous improvements in stability and performance for private cloud compute
Implement new functionality emerging from the research community
Write performant and scalable frameworks in Swift and C++
Distribute and coordinate ML inference tasks to different hardware acceleration IP blocks
Integrate inference code into a full service stack to ensure reliable and performant user traffic

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.