AI/ML Performance Modeling & Validation Engineer

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

Benefits offered by this job

Stock programs

Medical and dental coverage

Tuition reimbursement

Relocation assistance

Job summary

Apple Inc. in Cupertino, California, seeks an AI/ML Performance Analysis and Validation Engineer to build models that test future products and project performance on Apple Silicon.

You will validate critical use cases and reason about performance, collaborating with product, hardware, software, firmware, and data-center teams. The role requires deep computer-architecture knowledge, strong Python skills, and experience with ML frameworks (PyTorch, JAX, CoreML).

Qualifications

• Technical BS/MS/PhD or equivalent degree in Computer Science or related field.
• 3+ years of proven industry or research experience in building performance models and validation of AI/ML applications on dedicated accelerators.
• Deep understanding of Computer Architecture including processing units, interconnects and memory systems.
• Strong ability to build models of applications and platforms, characterize their performance and reason about them.
• Strong understanding of performance trade-offs in distributed computing systems.
• Excellent proficiency in Python and strong software engineering skills.
• Good understanding of machine learning fundamentals and familiarity with ML frameworks (PyTorch, JAX, CoreML).
• Strong data reasoning skills.
• Excellent communication and collaboration skills.

Responsibilities

• Build models of applications and platforms, and the tools used to test product viability.
• Analyze and validate critical use cases against performance goals.
• Analyze and reason about performance data.
• Tackle open-ended and challenging issues.
• Collaborate with cross-functional teams to define a vision for future products and present results.
• Help define technical requirements for Apple's future products.
• Stay current on AI/ML state of the art, SoC, and data center architectures.

Skills

Python

ML frameworks

Data reasoning

Communication skills

Software engineering

Computer architecture

Education

BS/MS/PhD in CS

Tools

PyTorch

JAX

CoreML

Job description

Apple Inc. in Cupertino, California, seeks an AI/ML Performance Analysis and Validation Engineer to build models that test future products and project performance on Apple Silicon.

You will validate critical use cases and reason about performance, collaborating with product, hardware, software, firmware, and data-center teams. The role requires deep computer-architecture knowledge, strong Python skills, and experience with ML frameworks (PyTorch, JAX, CoreML).