Machine Learning Engineer, Platform Architecture

In Apple’s Platform Architecture group, this Machine Learning Engineer role partners with engineers across Apple to map machine learning workloads onto Apple silicon. The work focuses on building optimized ML workload implementations and using performance modeling and simulation to guide and validate architecture decisions.

Responsibilities

• Create optimized implementations of ML workloads on Apple silicon, including Neural Engine, GPU, and CPU.
• Collaborate with IP and SoC architecture teams to build performance models and simulations for future hardware.
• Run performance studies that support and validate architecture decisions.
• Work with system teams to develop high-level performance models for emerging ML techniques and analyze architecture trade-offs.
• Investigate multiple approaches for mapping ML workloads to Apple silicon, then develop the corresponding performance models and simulations.
• Share findings with software and algorithm teams.

Requirements

• Bachelor’s degree.
• Programming experience in C/C++ and/or Python.
• Experience in computer architecture.
• Experience with at least one hardware IP, such as ML hardware accelerators or processing units including GPU, image/video, CPUs, or similar.

Technologies

• C/C++
• Python
• Neural Engine
• GPU
• CPU
• PyTorch
• Compiler frameworks/technologies

Preferred Qualifications

• MS or PhD in EE/CE/CS or a related field, plus 10+ years of relevant experience.
• Experience with ML frameworks (for example, PyTorch) and efficient implementations of machine learning algorithms.
• Experience optimizing and deploying ML models and/or runtime frameworks in production inference and training environments.
• Experience designing experiments to evaluate ML model optimization techniques.
• Experience creating SoC or IP performance models and simulations.
• Verbal and written communication skills for collaboration with partner teams.
• Ability to prototype algorithms on CPU, GPU, and Neural Engine, analyze performance metrics, and build high-level complexity models.
• Understanding of compiler frameworks/technologies.

Location and Compensation

Cupertino, CA (onsite). Salary range: USD 184,700 - 277,600 per year.

Benefits

• Comprehensive medical and dental coverage
• Retirement benefits
• Discounted products and free services
• Reimbursement for certain educational expenses, including tuition
• Discretionary bonuses or commission payments and relocation (if eligible)
• Participation in Apple’s discretionary employee stock programs
• Eligibility for discretionary restricted stock unit awards
• Ability to purchase Apple stock at a discount through the Employee Stock Purchase Plan (voluntary)