Staff/Principal Engineer - AI/ML & System-Level Validation

About the role

Principal Member of Technical Staff — ROCm Validation & System-Level Quality About the Role We are seeking a Principal Member of Technical Staff (PMTS) to serve as the senior technical leader for ROCm software validation across compute workloads and server-class systems.

What you'll do

• Do Own the end-to-end validation architecture for ROCm — unit, integration, framework, workload, performance, stress, stability, scale-out
• Lead system-level testing for server nodes — multi-GPU topologies, PCIe/Infinity Fabric/xGMI, BMC/IPMI, thermal/power, firmware interactions
• Drive compute workload validation and characterization — LLM training and inference (PyTorch, vLLM, Triton, JAX), recommender systems, scientific HPC kernels
• Architect the test infrastructure — distributed test runners, GitHub Actions / Jenkins / internal CI fleets, hardware lab orchestration, result data lakes
• Champion modern, agile quality engineering — shift-left testing, test pyramids, contract testing between layers, hermetic test environments, deterministic reproducers

What you need

• 12–22 yrs of experience
• Machine learning
• Python
• LLMs
• PyTorch
• Deep learning
• TensorFlow
• B.Sc or M.Sc or PhD or B.Tech or B.E.

Good to have

• CI/CD
• Git
• Test automation
• C++
• Test cases
• Jenkins