AMD ยท 1 week ago
Senior Staff Software Development Engineer
San Jose, CA
Full-time
Hybrid
Senior Level, Lead/Staff
$192K/yr - $288K/yrAMD is a leading company in next-generation computing experiences, and they are seeking a Senior Staff Software Development Engineer to provide technical leadership in the development of AI frameworks. The role involves optimizing deep learning frameworks for AMD GPUs and collaborating with various engineering teams to enhance performance and integration.
AI InfrastructureArtificial Intelligence (AI)Cloud ComputingComputerEmbedded SystemsGPUHardwareSemiconductor
Hiring Manager
Tressa Cooper (she/her)
Responsibilities
Optimize Deep Learning Frameworks: Enhance and optimize frameworks like PyTorch, vLLM, SGLang for AMD GPUs in open-source repositories
Develop GPU Kernels: Create and optimize GPU kernels to maximize performance for specific AI operations
Develop & Optimize Models: Design and optimize deep learning models using quantization specifically for AMD GPU performance
Collaborate with GPU Library Teams: Work closely with internal teams to analyze and improve training and inference performance on AMD GPUs
Collaborate with Open-Source Maintainers: Engage with framework maintainers to ensure code changes are aligned with requirements and integrated upstream
Software Engineering Best Practices: Apply sound engineering principles to ensure robust, maintainable solutions
Qualification
Required
Experience in developing and optimizing deep learning frameworks for AMD GPUs
Ability to engage with both internal GPU library teams and open-source maintainers
Experience in optimizing frameworks like PyTorch, vLLM, SGLang for AMD GPUs in open-source repositories
Ability to create and optimize GPU kernels to maximize performance for specific AI operations
Experience in designing and optimizing deep learning models using quantization specifically for AMD GPU performance
Ability to collaborate closely with internal teams to analyze and improve training and inference performance on AMD GPUs
Ability to engage with framework maintainers to ensure code changes are aligned with requirements and integrated upstream
Application of sound engineering principles to ensure robust, maintainable solutions
BS, MS or PhD in Computer Science, Computer Engineering, Electrical Engineering, or related technical fields
Preferred
Experienced in designing and optimizing GPU kernels for deep learning on AMD GPUs using HIP, CUDA, and assembly (ASM)
Strong knowledge of AMD architectures (GCN, RDNA) and low-level programming to maximize performance for AI operations
Experience with AI software framework, such as PyTorch, vLLM, SGLang, benchmarking and profiling
Experience using profiling and benchmark tooling for large models
Experience with model optimization, such as low-precision quantization (MXFP4, FP8, INT4), sparsity
Solid understanding of model architectures, LLMs, MoE, diffusion
Proficient in C++ programming
Experience developing and debugging in Python
Team player and ready to work with a geographically distributed team
Benefits
AMD benefits at a glance.
Company
AMD
Advanced Micro Devices is a semiconductor company that designs and develops graphics units, processors, and media solutions.
Founded in 1969
10001+ employees
H1B Sponsorship
AMD has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (836)
2024 (770)
2023 (551)
2022 (739)
2021 (519)
2020 (547)
Funding
Current Stage
Public CompanyTotal Funding
unknownKey Investors
OpenAIDaniel Loeb
2025-10-06Post Ipo Equity
2023-03-02Post Ipo Equity
2021-06-29Post Ipo Equity
Leadership Team
Lisa Su
Chair & CEO
Mark Papermaster
CTO and EVP
Recent News
2026-01-19
2026-01-18
2026-01-17
Company data provided by crunchbase