✦ Luna Orbit — AI & Machine Learning

AI Solution Integration Principal Engineer (SW)

at Advanced Micro Devices

📍 Austin, Texas, United States Hybrid Posted April 02, 2026
Type Full-Time
Experience lead
Exp. Years 5+ years
Education Bachelor's, Master's, or PhD in Computer Engineering, Computer Science, Electrical Engineering, or a related technical field
Category AI & Machine Learning

AMD is seeking an AI Solution Integration Principal Engineer to lead customer-facing, full-stack debugging of AI workloads on AMD GPUs. This hands-on role spans ROCm software, GPU microarchitecture, and AI frameworks to optimize performance and production readiness.

  • Own end-to-end integration and full-stack debugging of AI workloads across customer environments; serve as escalation point for production-blocking issues; diagnose ROCm stack issues; drive workload tuning and scaling; mentor senior engineers

Role encompasses end-to-end AI workload integration, including ROCm stack, runtime, compilers, and libraries, with emphasis on performance modeling, observability, and collaboration with silicon and firmware teams.

The ideal candidate is a senior AI solution integration engineer with broad experience across AI frameworks, ROCm software, and GPU microarchitecture. They can lead full-stack debugging and optimization for AI workloads on AMD hardware, while communicating effectively with customers and mentoring others.

Bachelor'sMaster'sor PhD in Computer EngineeringComputer ScienceElectrical Engineeringor relatedHands-on AI technology experienceSenior technologist with ROCm/ROCm experience
Experience across ROCmLinux-based GPU systemsInfluence on GPU/firmware/platform architectureExperience with LLMsdistributed traininglarge-scale inference
ROCmLinuxGPU driversFirmware
LinuxPythonML frameworks (PyTorch)ROCmHIP runtimeROCm stackMLIRLLVMC++CUDAdata center GPUAI workload optimizationdistributed traininginference workloads
LinuxPythonMachine LearningROCmROCm softwareGPUsROCmGPU microarchitectureROCm toolsCUDA
Customer facingTechnical communicationMentoring
Industry Semiconductors
Job Function Lead full-stack AI workload integration and optimization for AMD GPU platforms
Role Subtype ML Systems Engineer
Tech Domains Linux, Python, Java, JavaScript, SQL / PostgreSQL
AI Customer EngineeringROCmROCm softwareLinux-based GPU systemsGPU microarchitectureAI workload optimizationdata center GPUFCC framework PyTorch Triton MegatronLM vLLMTritonMegatronLMvLLMPyTorchHIP runtimeROCm librariesHIPobservability toolingLinuxPythonGPUNVIDIA or AMD GPUML frameworks

Must have a Master's or PhD in Computer Engineering, Computer Science, Electrical Engineering, or related

Apply for this Position →

Get matched to jobs like this

Luna finds roles that fit your skills and career goals — no endless scrolling required.

Create a Free Profile