Sr. Software Development Engineer (GPU Machine Learning Performance)
- Discuss this with your agent
- Open in Claude
- Open in ChatGPT
Role details
Tech stack
Job description
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. 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. Responsibilities: THE ROLE: AMDâs Advanced Technology Group is an entrepreneurial research and development team to build AMDâs future advanced platforms & products. Our teams work closely with outside customers and internal teams to develop hardware, software, and systems solutions into next generation computing platforms. As part of AMD Advanced Technology Group, you will have the opportunity to be part of a winning team that will collaborate with internal teams & customers, explore new platform technologies, and lead the development of best-in-class hardware, software, and systems technologies that our customers will use for real-world problems. In this role, you will join our team working on AMDâs next generation GPU edge inference solutions, ensuring machine learning solutions hit performance goals through GPU performance analysis, profiling, workload optimization and other duties to unlock powerful experiences for end-users with AMD devices., AMD is looking for a seasoned Principle Staff Member, to join our Advanced Technology Group GPU based Machine Learning development to work on AMDâs long-term Graphics and Machine Learning solutions, which will have significant impact for AMD, and the future of our products. Our group works on forward-looking projects and novel system concepts that significantly impact AMDâs future product portfolio. Your role will contribute to shaping AMDâs forward-looking hardware and software technologies., * Work within a team on Machine Learning SW and Workload development in pre and post-silicon phases of product development enabling next-generation of ML workloads in functional and performance attainment.
- Profile end-to-end inference pipelines to identify performance bottlenecks spanning kernels, scheduling, memory behavior, and model execution logic.
- Optimize execution paths for transformer models, including attention computation, KV-cache behavior, and efficient memory layouts.
- Develop and refine strategies for batching, parallelism, quantization, and distributed execution across single-node and multi-node environments.
- Collaborate with compiler, kernel and user mode driver teams, and hardware architecture teams to improve system-level efficiency and remove scaling blockers.
- Build performance models, benchmarking suites, and automated evaluation tools to quantify gains and track regressions.
- Shape best practices for profiling, debugging, and performance tuning across engineering teams.
- Communicate results, insights, and recommendations clearly to both technical and non-technical partners.
- Contribute to AMDâs next-generation GPU Machine Learning technologies.
Requirements
- Strong experience in high-performance GPU computing, inference optimization, or systems-level performance engineering.
- Deep understanding of machine learning architectures and large-scale inference behavior.
- Demonstrated success in building, optimizing, Machine Learning kernels, workloads, models for performance and power.
- Proficiency in GPU profiling tools, memory analysis, and performance tracing (e.g., HIP/CUDA profiling tools, custom instrumentation).
- Ability to reason about compute/memory tradeoffs, kernel scheduling, KV-cache patterns, and parallel execution models.
- Hands-on experience with GPU kernels, inference runtimes, or compiler-adjacent optimizations.
- Demonstrated track record of delivering measurable performance improvements in complex systems.
- Ability to drive technical direction in ambiguous or rapidly evolving problem spaces.
- Track record in using AI tooling to accelerate engineering work.
- Strong collaboration and communication skills., * Master / Bachelor of Computer Systems, Computer Science, Software Engineering, or similar degree.
About the company
AMD does not accept unsolicited resumes from headhunters, recruitment agencies, or fee-based recruitment services. AMD and its subsidiaries are equal opportunity, inclusive employers 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 here. This posting is for an existing vacancy.
Apply for this position
This job is hosted externally. Click below to view the full posting and apply.
Apply on indeed.comGood distractions
Talks and stories from around this role â technically off-topic, practically not.
Moments
Explore playlistsVideos
See allRelated articles
See all
MLOps And AI Driven Development
Stephan Gillich - Bringing AI Everywhere
Dev Digest 157: CUDA in Python, Gemini Code Assist and Back-dooring LLMs
Got AI ideas but no money? Here are 10 free ways to level up your AI skills with Google Cloud