JobsPrincipal AI Performance Engineer
Job description
AMD is seeking a Principal AI Performance Engineer to optimize AI inference performance on AMD GPUs. This role involves leading a technical team to tackle complex performance challenges across the stack, from profiling to optimization. The engineer will work closely with customers to deliver measurable performance improvements and develop methodologies that enhance team capabilities. The position requires a deep understanding of GPU performance and AI workloads, with a focus on innovation and collaboration.
Requirements
- 7+ years of software development experience in GPU computing, AI systems, or high-performance computing
- Deep hands-on experience with AI serving frameworks and their internals
- Strong background in end-to-end workload profiling and bottleneck diagnosis
- Understanding of GPU kernel performance characteristics
- Ability to read and reason about kernel-level profiling data
- Understanding of model architectures and inference paradigms
- Experience with custom kernel development or integration
- Understanding of multi-GPU and multi-node distributed systems
- Strong proficiency in Python and C++
- Customer-facing technical leadership experience
- Fluent in AI-assisted development
- Strong Linux systems knowledge
- Excellent written and verbal English communication skills
- Bachelor's, Master's, or PhD in Computer Science, Computer Engineering, Electrical Engineering, or equivalent
Responsibilities
- Drive performance optimization end-to-end across the stack on leading models and customer-relevant serving configurations
- Profile, diagnose, and resolve the hardest cross-stack performance bottlenecks
- Diagnose kernel-level performance issues using profiling tools and translate findings into actionable optimizations
- Lead customer-facing technical engagements and present findings
- Integrate and optimize custom kernels within serving frameworks
- Optimize multi-node distributed inference and develop performance optimization methodologies
- Leverage AI agents to accelerate daily work and define best practices
- Upstream optimizations into open-source frameworks
Benefits
- AMD provides a competitive 'Total Rewards' package that focuses on financial growth, health, and work-life balance.
Is this posting expired or inaccurate?
