Machine Learning Engineer, Platform Architecture

Job Description

Join Apple's Platform Architecture group to create optimized implementations of ML workloads on Apple silicon and collaborate with various teams to develop performance models and simulations for future hardware.

Requirements

• Bachelor's degree
• Ability to program in C/C++ and/or Python
• Knowledge of computer architecture fundamentals
• Domain knowledge in at least one hardware IP: ML HW accelerators or processing units such as GPU, image/video, CPUs, or similar

Responsibilities

• Create optimized implementations of ML workloads on Apple silicon including Neural Engine, GPU, and CPU
• Collaborate with IP and SoC architecture teams to develop performance models and simulations of future hardware
• Conduct performance studies to inform and validate architecture decisions
• Collaborate with system team to create high-level performance models of emerging ML techniques and analyze system architecture trade-offs

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.