Hardware Performance Engineer Jobs

1 job found

Today

Machine Learning Engineer, Platform Architecture

Apple Cupertino, CA

Cupertino, CA

Today

$185k - $278k/yr Full time Ai Ml Apple Silicon Artificial Intelligence

Apply Now

Job Description

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)

View full job page →

Get Hardware Performance Engineer Jobs alerts

New Hardware Performance Engineer Jobs, daily.