AI/ML Performance Engineer — Apple Silicon

Apple Inc.; Cupertino, Northern (CA, KY); Hybrid; USD 185,000 - 325,000; Full time

Job summary

Apple Inc. in Cupertino seeks a performance-focused engineer to analyze workloads, identify bottlenecks, and propose hardware/software improvements for Apple Silicon.

You will implement optimized ML workloads on Apple Silicon, collaborate with system teams to model performance, and contribute optimized libraries and tooling for AI/ML applications.

This role requires strong C/C++ and Python skills and experience in performance optimization across CPU, GPU, and neural accelerators.

Qualifications

• Bachelor's degree or equivalent in Computer Engineering, Computer Science, or a related field.
• Experience in software development for AI/ML HW accelerators, GPUs processing units, image/video encoders, or similar.
• Experience with AI/ML, graphics, or HPC performance benchmarks and workloads.
• MS/PhD in Computer Science, Computer Engineering, Electrical Engineering, or related field preferred.

Responsibilities

• Conduct performance studies to inform and validate architecture decisions.
• Create optimized implementations of machine learning workloads on Apple Silicon, including Neural Engine, GPU, and CPU.
• Collaborate with system teams to create performance models of emerging AI/ML techniques and analyze system architecture trade-offs.
• Work with software development tools teams to deliver performance analysis instruments and optimized libraries and frameworks for AI/ML applications.

Skills

C/C++ programming

Python scripting

AI/ML performance

Graphics / HPC benchmarks

Education

Bachelors in Computer Engineering or Computer Science

MS/PhD in CS/CE/EE

Tools

PyTorch

Job description

Apple Inc. in Cupertino seeks a performance-focused engineer to analyze workloads, identify bottlenecks, and propose hardware/software improvements for Apple Silicon.

You will implement optimized ML workloads on Apple Silicon, collaborate with system teams to model performance, and contribute optimized libraries and tooling for AI/ML applications.

This role requires strong C/C++ and Python skills and experience in performance optimization across CPU, GPU, and neural accelerators.