AMD ยท 3 months ago
Senior Staff Software Development Engineer- GPU/AI/ML
Santa Clara, California
Full-time
Onsite
Senior Level, Lead/StaffAdvanced Micro Devices, Inc (AMD) is dedicated to transforming lives with technology and building products that accelerate next-generation computing experiences. The Senior Staff Software Development Engineer will play a crucial role in improving the performance of AI applications and benchmarks, focusing on optimizing software for AMD's hardware and driving innovation in AI systems.
AI InfrastructureArtificial Intelligence (AI)Cloud ComputingComputerEmbedded SystemsGPUHardwareSemiconductor
Responsibilities
Architect and Drive the AI Software Stack: You will establish best practices and optimize performance from the lowest-level GPU kernels to large-scale distributed systems, shaping the foundational software for AMD hardware. By leveraging cutting-edge Large Language Models (LLMs) and agent-based technologies, you will accelerate the development and performance enhancement of the AMD ROCm ecosystem, ensuring it remains at the forefront of AI innovation
Accelerate Foundational Models: Your work will directly accelerate cutting-edge applications like foundation models (LLMs) and autonomous AI agents, ensuring AMD is the platform of choice for the most demanding workloads
Innovate Across Hardware and Software: You will contribute to the entire co-design lifecycle, from influencing future GPU architectures to developing groundbreaking software for new accelerators and collaborating with the broader AI community
Qualification
Required
Exceptional technical expertise in high-performance C++ software engineering
Proficiency in low-level GPU programming
Robust understanding of Large Language Models (LLMs) and AI systems
Experience in kernel engineering with 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
Experience contributing to the ROCm ecosystem (e.g., rocBLAS, hipDNN, Composable Kernel, AITemplate)
Experience with CUDA libraries (e.g., cuBLAS, cuDNN, CUTLASS, Thrust, CUB, NCCL)
Experience with the C++/HIP/CUDA core of ML frameworks like PyTorch, TensorFlow, or JAX
Deep understanding of LLMs, including transformer architectures and attention mechanisms
Hands-on experience in advanced model alignment and post-training techniques like Supervised Fine-Tuning (SFT) and Reinforcement Learning
Familiarity with cutting-edge trends in LLM advancements
Bachelor's degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent
Preferred
Lengthy professional software development experience in performance-critical environments
Extensive hands-on experience in GPU programming (HIP/CUDA) and optimizing deep learning kernels and operators
Fundamental understanding of GPU architecture and memory hierarchy
Expert-level proficiency in modern C++ and object-oriented design
Deep experience using GPU profiling and performance analysis tools (e.g., AMD ROCm Profiler, NVIDIA Nsight)
Deep knowledge of transformer architectures, attention mechanisms, and modern AI systems (Generative AI, Agentic AI)
Hands-on experience optimizing the post-training and inference pipelines of Large Language Models (LLMs)
Strong technical ownership, communication, and problem-solving skills
Experience or deep expertise with the AMD ROCm/HIP ecosystem
Benefits
Benefits offered are described: [AMD benefits at a glance](https://amd.jibeapply.com/benefits).
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
GlobeNewswire
2026-01-09
Company data provided by crunchbase