> Markdown version of [/jobs/ext/2038966-principal-software-development-engineer-ai-ml](https://www.wearedevelopers.com/jobs/ext/2038966-principal-software-development-engineer-ai-ml). 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 Software Development Engineer - AI/ML - **Company:** Advanced Micro Devices, Inc. - **Location:** San Jose, CA, United States - **Salary:** $240,000.0 - **Contract:** Permanent contract - **Skills:** Artificial Intelligence, C++ (Programming Language), Profiling, Nvidia CUDA, Computer Engineering, Software Debugging, Revision Control Systems, Machine Learning, Object-Oriented Software Development, Performance Tuning, Tensorflow, Software Engineering, Software Systems, Reinforcement Learning, Graphics Processing Unit (GPU), Parallel Computation, Gpu Programming, Git, Linux Development, Information Technology, ONNX (Open Neural Network Exchange) Format - **Published:** August 12, 2026 - **Apply:** https://diversityjobs.com/main/sendform/8/8/28176/1/17905722?backUrl=%2Fcareer%2F17905722%2FPrincipal-Software-Development-Engineer-Ai-Ml-California-San-Jose ## About the Role You are passionate about building high-performance AI software and solving complex optimization challenges. You possess strong compiler fundamentals, GPU programming expertise, and a proven ability to improve performance for large-scale AI models. You enjoy collaborating with architecture, compiler, runtime, and machine learning teams to deliver innovative solutions that scale across current and future AMD GPU platforms. You have a strong sense of ownership, excellent problem-solving skills, and a track record of driving meaningful performance improvements across software stacks., * Strong proficiency in C++ and object-oriented software development. * Hands-on experience with GPU programming technologies such as HIP, CUDA, Triton, or equivalent parallel programming frameworks. * Experience with compiler technologies including MLIR, LLVM, or related compiler infrastructures. * Strong understanding of machine learning frameworks, model execution pipelines, and AI workload optimization. * Experience profiling and optimizing performance-critical software systems. * Proficiency with Linux development environments, source control systems (Git), debugging tools, and performance profilers. * Strong analytical, communication, and problem-solving skills., * Bachelor's or Master's degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent ## Description We are seeking a highly motivated and technically exceptional PMTS AI/ML Compiler Engineer to join AMD's AI Software organization. In this role, you will drive compiler and performance optimizations that enable state-of-the-art AI training and inference workloads on AMD GPU platforms. You will work at the intersection of compiler technology, machine learning frameworks, GPU architecture, and performance engineering, contributing to the ROCm software ecosystem and ONNX Runtime execution stack. The ideal candidate combines deep compiler expertise with hands-on experience optimizing AI workloads, and thrives in a fast-paced environment collaborating across architecture, kernel, runtime, and framework teams., * Work with AMD's architecture specialists to improve future products * Design, develop, and optimize compiler transformations and operator lowerings within MLIR, LLVM, and the ROCm software stack. * Develop scalable solutions that improve portability and performance across AMD GPU architectures. * Collaborate with compiler and runtime teams to develop advanced optimization strategies for AI workloads. * Implement, validate, and maintain ONNX operators and microbenchmark test suites. * Lead performance optimization efforts for AI training, fine-tuning, reinforcement learning (RL), and inference workloads on AMD GPUs. * Design and implement graph-level, kernel-level, and runtime-level optimizations that deliver measurable performance gains. * Analyze workload execution characteristics and identify opportunities for improvements in compute efficiency, memory utilization, and communication overhead. * Drive performance characterization and benchmarking across pre-silicon and post-silicon environments., AMD may use Artificial Intelligence to help screen, assess or select applicants for this position. AMD's "Responsible AI Policy" is available here. ## Related Videos - [How a Small Team Shrank a Microsoft Monorepo by 94%](https://www.wearedevelopers.com/videos/1236-how-a-small-team-shrank-a-microsoft-monorepo-by-94) - [Profiling Symfony & PHP apps with Blackfire](https://www.wearedevelopers.com/videos/265-profiling-symfony-php-apps-with-blackfire) - [Machine learning in the browser with TensorFlowjs](https://www.wearedevelopers.com/videos/155-machine-learning-in-the-browser-with-tensorflowjs) - [Coffee with Developers - Stephen Jones - NVIDIA](https://www.wearedevelopers.com/videos/1303-coffee-with-developers-stephen-jones-nvidia) - [Enhancing Workload Security in Kubernetes](https://www.wearedevelopers.com/videos/356-enhancing-workload-security-in-kubernetes) - [Git for Code Reviews](https://www.wearedevelopers.com/videos/429-git-for-code-reviews) ## Related Articles - [MLOps And AI Driven Development](https://www.wearedevelopers.com/magazine/82-mlops-and-ai-driven-development) - [How to Become an AI Engineer](https://www.wearedevelopers.com/magazine/331-how-to-become-an-ai-engineer) - [Stephan Gillich - Bringing AI Everywhere](https://www.wearedevelopers.com/magazine/489-stephan-gillich-bringing-ai-everywhere) - [Transforming Software Development: The Role of AI and Developer Tools](https://www.wearedevelopers.com/magazine/527-transforming-software-development-the-role-of-ai-and-developer-tools) - [Dev Digest 157: CUDA in Python, Gemini Code Assist and Back-dooring LLMs](https://www.wearedevelopers.com/magazine/557-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](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)