Machine Learning Engineer, Platform Architecture

Machine Learning Engineer position in Apple’s Platform Architecture group, focused on mapping ML workloads to Apple silicon and using performance modeling and simulations to guide architecture choices.

Responsibilities

• Develop optimized implementations of ML workloads on Apple silicon across Neural Engine, GPU, and CPU
• Work with IP and SoC architecture teams to build performance models and simulations for future hardware
• Run performance studies to support and validate architecture decisions
• Partner with system teams to create high-level performance models for emerging ML techniques and evaluate system architecture trade-offs

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 HW accelerators or processing units including GPU, image/video, CPUs, or similar

Preferred Qualifications

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

Technologies

• C/C++, Python
• PyTorch
• GPU, CPU, Neural Engine
• ML HW accelerators
• image/video
• Compiler frameworks/technologies
• Machine learning algorithms, ML frameworks

Location

• Cupertino, CA (onsite)

Salary

• USD 150,400 - 225,300 per year

Benefits

• Comprehensive medical and dental coverage
• Retirement benefits
• Range of discounted products and free services
• Reimbursement for certain educational expenses, including tuition
• Discretionary bonuses or commission payments, plus relocation
• Eligible for discretionary restricted stock unit awards
• Ability to purchase Apple stock at a discount with voluntary participation in the Employee Stock Purchase Plan