This role at Apple involves researching and developing advanced machine learning solutions focused on natural language processing, audio, computer vision, and multi-modal applications. The position requires collaboration with cross-functional partners to productionize large-scale ML solutions and provide technical guidance to improve workflows. The successful candidate will contribute to internal PyTorch model development frameworks and implement state-of-the-art neural network architectures. Mentorship of junior engineers and interns is also a key aspect of this role.

• Master's degree or foreign equivalent in Computer Science, Engineering, or a related field.
• 3 years of experience in the job offered or a related occupation.
• 2 years of experience applying ML fundamentals to improve solutions in ML systems, natural language processing, computer vision, or audio processing.
• 2 years of experience utilizing modern ML frameworks, including PyTorch or JAX, for training and evaluating generative AI models.
• 2 years of experience applying software engineering fundamentals to develop efficient and maintainable Python code for ML applications.
• 2 years of experience using collaboration tools including git and GitHub.

• Research and develop advanced machine learning solutions for NLP, audio, computer vision, and multi-modal applications.
• Work closely with cross-functional partners to productionize large-scale ML solutions.
• Provide technical guidance to improve workflows for training, evaluation, model optimization, and deployment.
• Contribute modular and performant Python code to internal PyTorch model development frameworks.
• Implement and maintain state-of-the-art neural network architectures in internal frameworks.
• Design, implement, and maintain production workflows for large-scale distributed training.
• Implement and support tooling for evaluating the quality of large deep learning models.
• Research and evaluate the latest techniques for machine learning model development.
• Mentor and guide junior engineers and interns in best practices for machine learning model development.

• 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.