AIML - Staff Machine Learning Engineer - Answers, Knowledge & Information

Job Description

Join the Answers, Knowledge & Information team at Apple to work on improving Siri and Apple products for millions of users by leveraging ML techniques.

Requirements

• MS in Computer Science or related field
• 5+ years of industry related experience, working in collaborate environments
• Utilizing: PyTorch, TensorFlow, and JAX for training and deploying deep learning models
• Understanding product requirements and then translating them into modeling tasks and engineering tasks
• Building machine-learned models for search relevance ranking, query understanding, and summarization

Responsibilities

• Shape upcoming products from Apple by leveraging applied machine learning techniques
• Work on understanding queries, rank user intents, rank documents, and generate answers to users’ questions
• Train and deploy models at scale using the latest advances for online inference optimization

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.