2 months ago
Responsibilities
- Conduct external benchmarking and research to identify GPU kernel performance opportunities.
- Analyze and optimize GPU kernels across the AI software stack.
- Partner with customers and hyperscalers to understand workload requirements and deliver architectural and software optimizations.
- Collaborate with hardware architecture, compiler, and framework teams to influence future silicon features.
- Improve ROCm applications, libraries, tools, and hardware through collaboration with AMD experts and external partners.
Requirements
- Strong understanding of Transformer, Attention, KV Cache, quantization, speculative decoding, and FlashAttention.
- Excellent GPU architecture knowledge, performance analysis, algorithm development, and machine learning experience.
- Proven history of optimizing GPU kernels with advanced algorithms across the AI software stack.
- Ability to drive cross-functional initiatives in fast-paced, ambiguous environments.
- Bachelor’s, master’s, or PhD degree in Computer Science, Computer Engineering, or equivalent.
Benefits
- Benefits are offered through AMD’s benefits program.
- Located in Austin, Texas.
- This role is not eligible for visa sponsorship.
Categories
About AMD
AMD designs and sells CPUs, GPUs, and adaptive/embedded computing products for PCs, data centers, gaming, and edge devices. Its portfolio includes Ryzen and EPYC processors, Radeon and Instinct graphics, and adaptive SoCs from its Xilinx acquisition, sold to OEMs, cloud providers, and device makers. Founded in 1969 and headquartered in Santa Clara, it is a public company on NASDAQ and supplies semi-custom chips for major game consoles.