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, specifically 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 like AWS, GCP, or Azure
• 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
• Engage in the full training lifecycle, including pre-training and mid-training approaches
• Address challenges in reward modeling and align models for various tasks
• Collaborate with cross-functional teams to integrate AI solutions into products
• Transform research prototypes into scalable production systems

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