2 months ago
Responsibilities
- Develop innovative techniques for Generative AI training and inference, including transformer architectures, distributed training parallelism, speculative decoding, and KV-cache optimization.
- Implement efficient Generative AI model architectures and demonstrate their benefits on AMD platforms.
- Integrate AMD-optimized models and libraries with open-source frameworks and publish training recipes.
- Collaborate with software and hardware teams to co-optimize end-to-end performance on current and future AMD solutions.
- Increase adoption of agentic workflows for optimizing and deploying Generative AI applications at scale.
- Publish and promote technical work at external venues, including major conferences.
- Collaborate with AMD researchers and external industry and academic partners to advance innovation on AMD platforms.
Requirements
- Deep technical expertise and hands-on experience with Generative AI model training and inference in areas such as LLMs, 3D World and Action Models, or image/video generation models.
- Experience training models at scale and developing efficient distributed training and inference approaches on AMD devices.
- Strong expertise in algorithmic innovation for efficient Generative AI applications across training and inference.
- Expertise or publications in efficient model architectures, optimized training, innovative parallelism strategies, or low-precision training.
- Experience productizing Generative AI models and training foundation models at scale.
- Familiarity with PyTorch, JAX, vLLM, SGLang, and MuJoCo.
- Publications presented at NeurIPS, CVPR, ECCV, ICCV, ICML, or ICLR are a plus.
- Excellent written, verbal, presentation, coordination, and collaboration skills.
- Several years of experience in AI, deep learning, and related software development.
- PhD or master’s degree in computer science, electrical engineering, mathematics, or a related field.
Benefits
- AMD describes benefits through its benefits-at-a-glance program.
- The role is based in San Jose, California, with Seattle, Washington, or Austin, Texas as possible alternative locations.
Tech Stack
Categories
AI ResearchML Engineering
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.