- San Jose, California, United States
- Senior
Lead end-to-end AI workload optimization strategy, driving performance across AMD’s AI software stack.
Responsibilities
- Set the technical vision and roadmap for workload optimization across the AI software stack
- Lead the profiling, analysis, and tuning of large-scale models to ensure out-of-the-box performance on AMD hardware
- Partner with top customers and hyperscalers to understand unique workload requirements and deliver tailored architectural wins and software optimizations
- Collaborate across hardware architecture, compiler, and framework teams to influence future silicon features
- Drive the development of advanced tools and frameworks for performance estimation, modeling, and automated reporting
Requirements
- PhD or Master’s degree in Computer Science, Electrical Engineering, or related field
- Nice to have: High-level technical leadership
- Nice to have: AI frameworks (PyTorch, JAX, vLLM, SGLang)
- Nice to have: ROCm software stack
- Nice to have: TorchProfiler