> Markdown version of [/jobs/ext/124429-principal-senior-gpu-software-performance-engineer-post-training](https://www.wearedevelopers.com/jobs/ext/124429-principal-senior-gpu-software-performance-engineer-post-training). Every page supports `.md` or `Accept: text/markdown`. Links point to the HTML versions so they work for humans too. Agent guide: [/agents.md](https://www.wearedevelopers.com/agents.md). --- # Principal/Senior GPU Software Performance Engineer - Post-Training - **Company:** Advanced Micro Devices, Inc. - **Location:** San Jose, CA, United States - **Experience:** Expert - **Contract:** Permanent contract - **Skills:** Computer Engineering, Distributed Systems, Python (Programming Language), Graphics Processing Unit (GPU), Pytorch, Deep Learning, Information Technology, Software Performance - **Published:** May 16, 2026 - **Apply:** https://www.nettemps.com/us/en/W4755F66C5ABEABF585.ntap ## About the Role * Proven GPU performance engineering for deep learning (ROCm/HIP, Triton, or similar). * Hands-on with SFT. LoRA and RL-based training at scale. * Strong PyTorch experience (torch.distributed, FSDP/ZeRO or equivalent). * Proficient in Python and C+; comfortable reading/writing kernels when needed. * Experience with distributed systems and collective communication libraries. * Track record of turning profiles into fixes, upstreaming changes, and documenting results. ACADEMIC CREDENTIALS * B.S./M.S./Ph.D. in Computer Science, Computer Engineering, Electrical Engineering, or equivalent ## Description * 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. ## Related Videos - [30 Golden Rules of Deep Learning Performance](https://www.wearedevelopers.com/videos/11-30-golden-rules-of-deep-learning-performance) - [Getting Started with Machine Learning](https://www.wearedevelopers.com/videos/260-getting-started-with-machine-learning) - [Photonic Computing: Programming a New Class of AI Accelerators (incl. Live Coding)](https://www.wearedevelopers.com/videos/100196-photonic-computing-programming-a-new-class-of-ai-accelerators-incl-live-coding) - [Best Practices for AI-Assisted Development of Distributed Systems](https://www.wearedevelopers.com/videos/100200-best-practices-for-ai-assisted-development-of-distributed-systems) - [Geometric deep learning for drug discovery](https://www.wearedevelopers.com/videos/264-geometric-deep-learning-for-drug-discovery) - [Accelerating Python on GPUs](https://www.wearedevelopers.com/videos/859-accelerating-python-on-gpus) ## Related Articles - [Highest Paying Tech Companies for Developers](https://www.wearedevelopers.com/magazine/220-highest-paying-tech-companies-for-developers) - [Got AI ideas but no money? Here are 10 free ways to level up your AI skills with Google Cloud](https://www.wearedevelopers.com/magazine/600-got-ai-ideas-but-no-money-here-are-10-free-ways-to-level-up-your-ai-skills-with-google-cloud) - [Dev Digest 120 - Apple and peers](https://www.wearedevelopers.com/magazine/455-dev-digest-120-apple-and-peers) - [Why Upskilling And Reskilling is Important For Developers](https://www.wearedevelopers.com/magazine/428-why-upskilling-and-reskilling-is-important-for-developers) - [How to Become an AI Engineer](https://www.wearedevelopers.com/magazine/331-how-to-become-an-ai-engineer) - [Résumé-Driven Development: How IT trends affect the job market for software developers](https://www.wearedevelopers.com/magazine/59-resume-driven-development-how-it-trends-affect-the-job-market-for-software-developers)