Advanced Micro Devices, Inc.-posted about 1 month ago
Full-time • Mid Level
Remote • Austin, TX
5,001-10,000 employees
Computer and Electronic Product Manufacturing

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. MTS SOFTWARE DEVELOPMENT ENGINEER THE ROLE: As a Member of Technical Staff in the Composable Kernel Library group in the GPU Kernel Engineering role, you will contribute to the development and optimization of GPU kernels with the AMD Radeon Open Ecosystem (ROCm). You will design and implement highly optimized primitives enabling flexible and efficient composition of kernels for deep learning operations such as GEMM, convolution, Attention, and Mixture of Experts. Collaboration with internal GPU library teams and open-source maintainers will be key to ensuring these components integrate seamlessly and deliver high performance. This position requires a solid understanding of GPU architectures and compiler technologies, strong engineering fundamentals, and a commitment to building scalable, maintainable solutions. THE PERSON: A skilled engineer with strong technical and analytical expertise in C++ development within Linux environments. The ideal candidate will thrive in both collaborative team settings and independent work, with the ability to define goals, manage development efforts, and deliver high-quality solutions. Strong problem-solving skills, a proactive approach, and a keen understanding of software engineering best practices are essential.

  • Works collaboratively across library teams to design, implement, validate, and deliver key software library capabilities, ensuring successful integration with GPU kernels into broader systems.
  • Provides significant technical inputs to the software roadmap.
  • Provides input to architecture, design, and documentation efforts as they arise across teams.
  • Works closely with senior engineering staff to guide technical execution within library teams on broad and complex technical challenges.
  • Works closely with senior engineering staff to guide technical execution within library teams on broad and complex technical challenges.
  • Collaborates with principal members of staff to develop the best technical designs and approaches that holistically consider constraints, performance, and roadmap commitments.
  • Executes, and report progress against project plans and delivery commitments.
  • Proactively identifies gaps and inconsistencies at project scope and develops plans to remediate in collaboration with library owners and engineering principal colleagues.
  • Engages in mentorships with senior and principal staff and champion internal best-practice as part of the regular software development lifecycle in an agile environment.
  • Hands-on experience with C++, with relevant applied experience using CUDA, HIP, OpenMP, MPI, or OpenCL for accelerated computing on CPUs and GPUs. Familiarity with other programming languages like Python and AI/ML Frameworks.
  • Experience contributing to technical and high-performance software systems supported by continuous improvement, regression and performance management systems, and software deployment and orchestration management (Jenkins/Docker/Kubernetes/numerical methods and analysis).
  • Experience with open-source software development including collaboration with community maintainers and submitting contributions, and various lifecycle methodologies such as Agile, Scrum, Kanban.
  • Knowledge of professional software engineering practices and best practices for the full software development life cycle including requirements elicitation and analysis, scoping/estimation, coding standards, code reviews, version control, build processes, testing, and continuous integration.
© 2024 Teal Labs, Inc
Privacy PolicyTerms of Service