Senior Principal AI Performance Engineer

About the role

Are you passionate about optimizing AI workloads and delivering real-world performance improvements on edge devices? We’re looking for an experienced engineer to help customers achieve best-in-class inference performance for production AI models running on Arm technology. This role is based in San Jose, with significant time spent working directly with customers across the Bay Area. Please note this role does not meet the eligibility requirements for sponsorship, and therefore the successful candidate must have the right to work in the US without relying on sponsorship by Arm.

Responsibilities

• Develop highly optimized solutions for AI workloads, from kernel level to system level, to meet the needs of the customer application.
• Create production quality reference implementations, documentation, and performance focused technical content
• Act as a technical bridge between customers and internal teams, driving resolution of complex performance issues
• Influence Arm’s IP and software roadmap through insights gained from real-world customer use cases

Skills

• Experience optimizing DNNs in Triton, CUDA or other kernel level programming language
• Deep understanding of parallel computing, memory hierarchies and performance optimization techniques for DNNs
• Strong programming skills in Python and C++, a solid experience with modern AI frameworks and execution models, experience with profiling and analysis tools
• Strong communication and interpersonal skills

Qualifications

• Experience in a customer facing or field engineering environment
• Experience within the Arm ecosystem
• Background in AI performance optimization for edge devices

Pay

Salary Range: $323,000-$437,000 per year