Machine Learning Engineer, Platform Architecture

Apple, Inc.; Cupertino (CA); On-site; USD 185,000 - 278,000; Full time

Benefits offered by this job

Medical and dental Coverage

Retirement benefits

Employee stock programs

Relocation assistance

Tuition reimbursement

Discounted products

Job summary

Apple’s Platform Architecture group in Cupertino is seeking an engineer to map ML workloads to Apple silicon. This onsite role collaborates with cross-team engineers to build optimized ML workload implementations and to develop performance models and simulations that guide architecture decisions.

You will communicate findings to software and algorithm teams to validate trade-offs and improve execution efficiency, and you will prototype algorithms on CPU, GPU, and Neural Engine.

Qualifications

• Bachelor’s degree in a relevant field
• 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

Responsibilities

• 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

Skills

C/C++

Python

Verbal and written communication

Performance modeling

ML frameworks

Education

Bachelor’s degree

MS or PhD in EE/CE/CS or related field

Tools

PyTorch

Compiler frameworks/technologies

Neural Engine

GPU

CPU

Job description

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, PyTorch) 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
• PyTorch
• Compiler frameworks/technologies

Compensation and benefits

Salary: USD 184,700 - 277,600 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)