Principal AI Performance Engineer

About the role

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.

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.

Requirements

• 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.

Skills

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

Benefits

Joining Arm means stepping into a career-defining opportunity. You’ll occupy a central role in the company’s most critical initiatives. These initiatives build how Arm innovates, scales, and partners globally.

Arm’s hybrid approach to working is centered around flexibility, where we split our time between the office and other locations to get our work done. Within that framework, we empower groups and teams to determine their own particular hybrid working pattern, depending on the work and the team’s needs.

Pay

Salary range: $262,700-$355,400 per year.

Salary is only one component of Arm's total reward package, which will be shared with candidates during the recruitment and selection process.

This role does not meet the eligibility requirements for sponsorship; the successful candidate must have the right to work in the US without relying on sponsorship by Arm.