AMD - San Jose, CA
posted 4 months ago
At AMD, we are committed to transforming lives through our technology, enriching industries, communities, and the world. Our mission is to create exceptional products that accelerate next-generation computing experiences, serving as the foundation for data centers, artificial intelligence, PCs, gaming, and embedded systems. The culture at AMD is built on pushing the boundaries of innovation to tackle the world's most pressing challenges. We prioritize execution excellence while fostering a direct, humble, collaborative, and inclusive environment that values diverse perspectives. The AI Group at AMD is on the lookout for talented and motivated engineers and scientists to contribute to Generative AI inference solutions. In this role, you will delve into and enhance state-of-the-art research from both academia and industry, innovating in software development, model optimization, and compression algorithms for Generative AI applications, including large language models (LLMs), Stable Diffusion, and multi-modal models. As a member of the core AI Solutions team, you will enable Generative AI across a wide array of AMD products, including RyzenAI, XDNA/NPU, CPU, GPU, and embedded architectures. Your contributions will significantly shape and influence AMD's inference and training solutions across the software stack, from frameworks like PyTorch and ONNX to custom hardware such as NPU and GPU. If you are passionate about software engineering, keeping up with the latest literature, generating novel ideas, and leading their implementation through high-quality code, this role is ideal for you. You should possess a deep understanding of foundational AI model architectures, software development, and AI/ML frameworks such as PyTorch, along with expertise in quantization and sparsity algorithms. Demonstrated experience in efficiently mapping AI models onto GPUs and/or other hardware accelerators is essential. You will thrive in a dynamic, fast-paced development environment, showcasing excellent collaboration skills and a spirit of innovation while working with multiple geographically dispersed engineering teams. Your work will involve developing next-generation framework software and guiding other senior developers and domain experts.