Apple is seeking a highly motivated machine learning engineer to join the All-Scale Modeling team. This role focuses on developing end-to-end machine learning systems to address complex physics modeling challenges for Apple hardware. The engineer will optimize performance and efficiency of kernels while designing advanced systems for intricate tasks. Collaboration with world-class teams is a key aspect of this position.

• Strong machine learning background in algorithms and engineering.
• Proficiency in Python or C++ development.
• Strong analytical and problem-solving skills.
• Rich experience in deploying, experimenting, and benchmarking models.
• B.S., M.S., or Ph.D. degree in Computer Science, Electrical Engineering, or a related field.
• Proficiency in CUDA or Metal programming.
• Strong background in numerical methods and physics such as ME, EM, or CFD.

• Develop end-to-end machine learning systems to solve complex physics modeling problems.
• Optimize kernels for performance and efficiency.
• Design and implement advanced agentic systems for complex tasks.
• Document and share findings to influence roadmap decisions.
• Collaborate with world-class teams.

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