Senior Staff Software Development Engineer- GPU/AI/ML

Advanced Micro Devices, IncSanta Clara, CA
138d

About The Position

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.

Requirements

  • Exceptional technical expertise in high-performance C++ software engineering and low-level GPU programming.
  • Robust understanding of Large Language Models (LLMs) and AI systems.
  • Mastery in designing complex, scalable systems using modern C++.
  • Fundamental grasp of GPU architectures (HIP/CUDA), memory hierarchies, and kernel optimization.
  • Significant hands-on experience in large-scale C++/HIP/CUDA projects.
  • Deep understanding of LLMs, including transformer architectures and attention mechanisms.
  • Hands-on experience in advanced model alignment and post-training techniques.

Nice To Haves

  • Experience or deep expertise with the AMD ROCm/HIP ecosystem.
  • Experience and interest in code generation and/or self-improving LLMs.

Responsibilities

  • Architect and Drive the AI Software Stack: Establish best practices and optimize performance from the lowest-level GPU kernels to large-scale distributed systems.
  • Accelerate Foundational Models: Directly accelerate cutting-edge applications like foundation models (LLMs) and autonomous AI agents.
  • Innovate Across Hardware and Software: Contribute to the entire co-design lifecycle, influencing future GPU architectures and developing groundbreaking software.

Benefits

  • AMD benefits at a glance.
© 2024 Teal Labs, Inc
Privacy PolicyTerms of Service