Apple is seeking skilled Machine Learning Engineers to develop sophisticated foundation models that enhance intelligent features across its product ecosystem. The team focuses on transforming cutting-edge research into scalable, production-ready AI solutions optimized for Apple silicon. Engineers will work on the full training lifecycle, addressing challenges in reward modeling and aligning models for various tasks. This role is ideal for those passionate about solving complex problems where research and product development are intertwined.

• MS or PhD in Computer Science, Machine Learning, or a related technical field
• Expert-level programming skills in Python
• Proficiency in machine learning frameworks such as Jax, PyTorch, and TensorFlow
• Strong background in distributed training, model optimization, and machine learning infrastructure
• Experience with large-scale model training and deployment
• Familiarity with Kubernetes, Docker, and cloud platforms such as AWS, GCP, and Azure
• Experience with distributed computing frameworks
• Experience with foundation models and large language models
• Background in multi-modal AI systems
• Demonstrated ability to transform research prototypes into production systems
• Published research or significant contributions to open-source ML projects
• Understanding of on-device machine learning techniques

• Develop and optimize foundation models for Apple products
• Transform cutting-edge research into scalable AI solutions
• Work across the full training lifecycle of machine learning models
• Address challenges in reward modeling and model alignment
• Collaborate with cross-functional teams to integrate AI features into products
• Ensure models are optimized for privacy and performance on Apple silicon

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