Principal / Senior GPU SW Performance Engineer — Post‑Training
San Jose, CA$240k - $360kFull-timePosted 6mo ago
About the role
Drive post‑training performance for AMD Instinct GPUs, focusing on finetuning/RL training workloads, multi‑GPU/multi‑node scaling, and cross‑stack optimizations to deliver fast, stable, reproducible training pipelines on ROCm.
Responsibilities
- Lead performance for finetuning and RL training solutions on AMD GPUs.
- Improve throughput, memory efficiency, and stability across data, model, and optimizer steps.
- Optimize multi‑GPU/multi‑node training and communication patterns.
- Contribute efficient kernels/ops and targeted graph‑level optimizations.
- Profile, diagnose, and resolve bottlenecks using standard tooling; prevent regressions in CI.
- Ship reproducible pipelines and documentation adopted by internal teams and external developers.
- Collaborate with framework, compiler, and model teams to land durable improvements.
Qualifications
- B.S./M.S./Ph.D. in Computer Science, Computer Engineering, Electrical Engineering, or equivalent
Skills
Required:GPU performance engineering for deep learningPyTorchPythonC++Distributed systems
Nice to have:SFT, LoRA, RL‑based training at scale
Last updated Oct 15, 2026. Interested? Apply on the company site.
About AMD
together we advance_AI
AMD delivers leadership high-performance and adaptive computing solutions to advance data center AI, AI PCs, intelligent edge devices, gaming, & beyond.
AIData CenterHigh-Performance ComputingGamingEdge Devices
More open roles at AMD
- System Thermal Design Staff Engineer4d agoPenang, Malaysia · Engineering
- PD Engineer4d agoHybrid (Bangalore) · Engineering
- PD Engineer4d agoBangalore · Engineering
- Low-Power & CDC Verification Engineer4d agoSanta Clara, California · Engineering
- Technical Program Manager4d agoShanghai · Engineering