Edge AI Performance Engineer (Kernel & Optimization)

Arm; San Jose (CA); Hybrid; USD 263,000 - 355,000; Full time

Job summary

Arm in San Jose seeks an experienced engineer to optimize AI workloads and deliver best-in-class inference performance on Arm technology. You will collaborate across teams, translating complex challenges into clear insights for engineers and leadership.

The role focuses on kernel-level implementations, power/performance optimization, and close work with customers across the Bay Area. Strong coding and communication skills are essential.

Qualifications

• 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, profiling and analysis tools.
• Strong communication and interpersonal skills.

Responsibilities

• Develop highly optimized solutions for AI workloads from kernel to system level.
• Create production quality reference implementations, documentation, and performance-focused content.
• Act as a technical bridge between customers and internal teams to resolve performance issues.
• Influence Arm’s IP and software roadmap based on real-world customer insights.

Skills

DNN optimization (Triton/CUDA)

Parallel computing

Python & C++

Communication skills

Job description

Arm in San Jose seeks an experienced engineer to optimize AI workloads and deliver best-in-class inference performance on Arm technology. You will collaborate across teams, translating complex challenges into clear insights for engineers and leadership.

The role focuses on kernel-level implementations, power/performance optimization, and close work with customers across the Bay Area. Strong coding and communication skills are essential.