Advanced Micro Devices, Inc-posted 4 months ago
5,001-10,000 employees

At AMD, our mission is to build great products that accelerate next-generation computing experiences – from AI and data centers, to PCs, gaming and embedded systems. Grounded in a culture of innovation and collaboration, we believe real progress comes from bold ideas, human ingenuity and a shared passion to create something extraordinary. When you join AMD, you’ll discover the real differentiator is our culture. We push the limits of innovation to solve the world’s most important challenges—striving for execution excellence, while being direct, humble, collaborative, and inclusive of diverse perspectives. Join us as we shape the future of AI and beyond. Together, we advance your career. In this role, you will play an important part in helping align AMD's hardware designs with the needs of AI software today and in the future. Your responsibilities will encompass gaining a comprehensive understanding of Machine Learning networks, the important kernels that are used within them, and how best to map them to current and future hardware features. This hands-on experience will not only involve delving into the fundamentals of GPU hardware capabilities but also actively engaging with compiler teams, kernel teams and hardware design teams. As part of your role, you will have the chance to contribute to the development of various components within a sophisticated software stack, the workloads themselves, and ultimately to the hardware designs needed to make them better. Get ready to explore and contribute to the cutting edge of AI technology.

  • Understand, implement and optimize GPU kernels using cutting-edge compiler technology
  • Holistically understand and conduct benchmarking on prevalent AI model workloads, including tasks in vision, speech recognition, language processing and generative models
  • Collaborate with AMD’s architecture specialists to enhance future products
  • Stay abreast of software and hardware trends and innovations, particularly in algorithms and architecture, ensuring the integration of the latest advancements into our projects
  • Participating in new GPU bring ups for AI software, including prototyping in emulated environments
  • Debug and resolve existing issues and research alternative, more efficient ways to accomplish the same work
  • Cultivate technical relationships with peers and partners, fostering collaborative efforts and knowledge exchange
  • Strong object-oriented programming background, C/C++ preferred
  • Proficient in crafting high performance GPU kernels
  • Previous exposure to cutting edge machine learning models
  • Familiarity with modern source control systems such as GitHub
  • Excellent verbal communication and writing skills
  • Proven expertise in optimizing high performance GPU kernels
  • Prior hands-on experience using AI accelerators
  • Past experience working with hardware teams and design feedback
  • A collaborative team player with experience working in geographically dispersed teams
© 2024 Teal Labs, Inc
Privacy PolicyTerms of Service