AI/ML Performance Analysis and Validation Engineer

• This is an opportunity to shape the architecture of future AI/ML applications at Apple and influence the design of the next generation of hardware platforms.
• 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 performing validation of AI/ML applications running 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.
• Hands on experience with implementing AI/ML applications with state-of-the-art architectures such as CNN, transformer and diffusion models and analyzing their performance characteristics.
• Experience in modeling data center architectures.
• Experience in building simulation technologies.
• Experience collaborating in large cross functional projects.

None flagged.