Machine Learning Engineer, Platform Architecture

In Apple’s Platform Architecture group, you will help map machine learning workloads to Apple silicon. This onsite role in Cupertino supports cross-team collaboration with engineers across Apple, with a focus on building optimized ML workload implementations and developing performance models and simulations that guide architecture decisions. You will also communicate findings to software and algorithm teams to help validate trade-offs and improve execution efficiency.

What

you’ll do

• Create optimized implementations of ML workloads on Neural Engine
  , GPU
  , and CPU running on Apple silicon
• Partner with IP and SoC architecture teams to develop performance models and simulations for future hardware
• Run performance studies to inform and validate architecture decisions
• Work with system teams to build high-level performance models for emerging ML techniques and evaluate system architecture trade-offs
• Explore multiple approaches to mapping ML workloads onto Apple silicon and produce supporting performance models and simulations
• Share results and learnings 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 GPUs, image/video, CPUs, or similar

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,
  Py Torch ) and efficient implementations of machine learning algorithms
• Experience optimizing and deploying ML models and/or runtime frameworks in production inference and/or 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 create high-level complexity models
• Understanding of compiler frameworks/technologies

Tech you may work with

• C/C++,
  Python
• Neural Engine
  , GPU
  , CPU
• Py Torch
• Compiler frameworks/technologies

Compensation and benefits

Salary: USD 184, per year.

• Comprehensive medical and dental coverage
• Retirement benefits
• A range of 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
• Opportunity to purchase Apple stock at a discount through the Apple Employee Stock Purchase Plan (if you voluntarily participate)