13 days ago
San Jose, CA, USAStaff+
Base Salary
$263k - $355k/yr
Responsibilities
- Develop optimized kernel-level and system-level solutions for AI workloads on Arm technology.
- Create production-quality reference implementations, documentation, and performance-focused technical content.
- Act as a technical bridge between customers and internal engineering teams to resolve complex performance issues.
- Diagnose DNN performance challenges and use customer insights to influence Arm’s IP and software roadmaps.
- Present technical progress and recommendations to engineers and senior leadership.
Requirements
- Experience optimizing deep neural networks in Triton, CUDA, or another kernel-level programming language.
- Deep understanding of parallel computing, memory hierarchies, and DNN performance optimization techniques.
- Strong programming skills in Python and C++.
- Experience with modern AI frameworks, execution models, profiling, and analysis tools.
- Strong communication and interpersonal skills.
- Customer-facing or field engineering experience, Arm ecosystem experience, and edge-device AI performance optimization are preferred.
Benefits
- Hybrid working with team-determined patterns and flexibility subject to local legal, regulatory, and tax considerations.
- The role is based in San Jose and involves significant time working with customers across the Bay Area.
- Reasonable accommodations are available during the application, interview, and employment process.
- Competitive total rewards package; salary is one component of the offering.
Categories
About Arm
Arm’s foundational technology is defining the future of computing. A future built by the greatest technology ecosystem in the world. A future built on Arm. Arm is everywhere technology matters. Technology matters everywhere. Together, we’ll power every technology revolution moving forward, including cloud computing, automotive and autonomous systems, IoT, the metaverse, and beyond. Changing the world. Again. On Arm.