Principal Software Developer - GPU AI/HPC kernels

WHAT YOU DO AT AMD CHANGES EVERYTHING 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. Join us as we shape the future of AI and beyond. THE ROLE: AMD is looking for a principal software developer to join our growing team. As a key contributor you will be part of our ROCm GPU‑compute mathematical libraries team working on innovative hardware and software technologies. You will help deliver exceptional performance and feature enhancements via maintainable code development, optimizations/tuning, and collaboration. THE PERSON: The successful candidate would possess excellent software skills in C/C++, Python and has expert understanding of software performance and computer architecture. Must have strong desire and ability to solve problems, and be able to schedule, and work on tasks independently. Should be a great team player with excellent communication skills. KEY RESPONSIBILITIES: Develop software in C++, Python, HIP, assembly, and SOTA programming technologies to enable key mathematical operations on GPU Design GPU computational software libraries for AI, HPC applications Aid management in planning, and delivering industry‑leading software for current and future processors Supervise small development team Carry‑out performance optimizations and projections for important use‑cases to maximize hardware utilization Support development of programs to sustain seamless performance analysis, and performance/functional test coverage Identify and help resolve quality issues working closely with libraries development teams and other internal engineering teams PREFERRED EXPERIENCE: 10+ years professional software development experience Demonstrated capacity to technically lead and people manage junior to mid‑level developers Proficient in C/C++ & Python programming employing best software design practices GPU software development or validation involving HIP, CUDA, or OpenCL Experience with software libraries and API design Exposure to Matrix/Tensor operations and numerical work Software emulation to support FP numerical formats is a plus Experience in software performance estimations, optimizations and debugging Ability to closely interact with technical leads, developers, and test teams to maintain and release production software ACADEMIC CREDENTIALS: Master’s or Doctoral degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent THE ROLE (GPU SOFTWARE ARCHITECT): As GPU Software Architect, you will provide technical leadership at the intersection of GPU architecture, multi‑ASIC platform bring‑up, and software enablement for next‑generation GPU products. This is a "Software‑First" architecture role: you will reimagine and redefine the end‑to‑end software libraries lifecycle as it spans multiple ASICs to create a unified software fabric and process supporting development of software libraries on cutting‑edge hardware. You will serve as a bridging authority between software architecture and the hardware ecosystem, ensuring that architectural intent translates into working, performant, and scalable solutions for partnerships established with software libraries teams. This role is focused on leading a team focused on new GPUs and new product introductions, with accountability spanning early architecture definition, pre‑silicon modeling, multi‑ASIC bring‑up strategy, and software readiness for emerging platforms. KEY RESPONSIBILITIES: Architecture & Platform Leadership: Provide technical leadership for GPU architecture decisions with direct impact on multi‑ASIC platforms, interconnects, memory systems, and scalability. Translate architectural concepts into concrete platform requirements spanning ASIC, firmware, drivers, and software libraries. Multi‑ASIC Bring‑Up Strategy: Define and lead bring‑up strategies for new GPU platforms, partnering with silicon, systems, and software teams to identify risks early and drive mitigation plans from pre‑silicon through first silicon. Hardware‑Software Co‑Design: Drive hardware/software interface definition, ensuring architecture choices support and reflect the drive toward performance and quality. Influence firmware, driver, runtime, and performance software design to align with architectural intent. Early Silicon Enablement: Act as a technical escalation point during early silicon bring‑up, debugging complex cross‑layer issues spanning hardware, firmware, and software. Guide the creation of diagnostics, validation tools, and bring‑up workflows that scale across teams and products. Cross‑Functional Technical Leadership: Work across architecture, design, verification, drivers, performance libraries, and product teams to ensure alignment. Provide technical mentorship and review, raising the overall effectiveness of teams working on new GPU platforms. Knowledge Capture & Reuse: Capture lessons learned from new product bring‑up and translate them into reusable architecture patterns, best practices, and documentation. Leverages AI‑assisted software development tools to accelerate the design, implementation, review, and documentation of complex software libraries. Establishes best practices for responsible use of AI assistance, including validation, review, and traceability of generated code and technical artifacts. PREFERRED EXPERIENCE: Deep experience in GPU, accelerator, or SoC architecture, including memory systems, interconnects, and scalability considerations. History of technical leadership across distributed, cross‑functional engineering teams. Strong background in systems software, firmware, drivers, or performance software used to enable new silicon. Proven experience in hardware/software co‑design, including defining interfaces and debugging cross‑layer issues. Hands‑on programming experience in C/C++ and Python. Familiarity with low‑level debugging tools and workflows. Experience working with performance modeling, simulators, or early validation infrastructure. Applied experience using AI‑assisted coding tools in professional software engineering workflows, including code generation, refactoring, test creation, documentation, and design exploration. Advanced degree in Computer Engineering, Electrical Engineering, Computer Science, or equivalent practical experience. ACADEMIC CREDENTIALS: Advanced degrees, such as M.Sc., M.Eng., Ph.D. are preferred LOCATION: Austin, TX NOTE: This role is not eligible for visa sponsorship. EEO STATEMENT: AMD is an equal opportunity, inclusive employer and will consider all applicants without regard to age, ancestry, color, marital status, medical condition, mental or physical disability, national origin, race, religion, political and/or third‑party affiliation, sex, pregnancy, sexual orientation, gender identity, military or veteran status, or any other characteristic protected by law. We encourage applications from all qualified candidates and will accommodate applicants’ needs under the respective laws throughout all stages of the recruitment and selection process. AMD may use Artificial Intelligence to help screen, assess or select applicants for this position. AMD’s "Responsible AI Policy" is available. This posting is for an existing vacancy. #J-18808-Ljbffr