AI/ML Systems Performance Engineer

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

Benefits offered by this job

Stock programs

Comprehensive medical & dental

Educational reimbursement

Relocation assistance

Job summary

Apple, Inc. is seeking a performance-focused engineer for the Platform Architecture group in Cupertino. You will analyze workloads, identify bottlenecks, and recommend hardware/software improvements across Apple Silicon.

You will implement optimized ML workloads on Neural Engine, GPU, and CPU, and collaborate with cross-functional teams to model performance and trade-offs. A strong background in computer architecture and AI/ML tooling is required.

Qualifications

• Bachelor's degree or equivalent in Computer Engineering, Computer Science, or related field.
• Experience in software development for AI/ML HW accelerators, GPUs, or related hardware IPs.
• Experience with AI/ML benchmarks and workloads.

Responsibilities

• Conduct performance studies to inform architecture decisions.
• Create optimized implementations of ML workloads on Apple Silicon (Neural Engine, GPU, CPU).
• Collaborate with system teams to model performance and analyze trade-offs.
• Develop performance analysis tools and optimize libraries for AI/ML apps.

Skills

C/C++

Python

PyTorch

Computer architecture

OS internals

Compiler technologies

AI/ML benchmarks

Education

Bachelor's degree or equivalent

MS/PhD in CS/CE/EE

Job description

Apple, Inc. is seeking a performance-focused engineer for the Platform Architecture group in Cupertino. You will analyze workloads, identify bottlenecks, and recommend hardware/software improvements across Apple Silicon.

You will implement optimized ML workloads on Neural Engine, GPU, and CPU, and collaborate with cross-functional teams to model performance and trade-offs. A strong background in computer architecture and AI/ML tooling is required.