← All roles

Lead AI/ML Software Use Cases Validation Engineer

AMD· Engineering
Apply on company site
BangaloreFull-timePosted 1w ago

About the role

Lead AI/ML validation engineer responsible for end-to-end validation, benchmarking, and optimization of AI/ML compute stacks and Physical AI SDKs across Linux platforms, collaborating with architecture, compiler, runtime, and hardware teams to deliver production-quality AI solutions aligned to customer use cases.

Responsibilities

  • Lead validation and quality ownership of AI/ML compute stacks on Ubuntu and Yocto
  • Define validation strategy, test architecture, and coverage across functional, performance, stress, regression, and scalability testing
  • Own and drive the defect lifecycle, including triage, root cause analysis, and closure
  • Validate end-to-end AI pipelines, including model training, conversion, and optimization (e.g., PyTorch → ONNX)
  • Kernel execution, memory transfers, and inference accuracy validation
  • Define and execute AI benchmarking and profiling strategies for training and inference workloads
  • Analyze compute, memory, and latency bottlenecks and drive system-level and model-level optimizations
  • Validate AI frameworks and runtimes (PyTorch, TensorFlow, ONNX Runtime)
  • Execute and optimize workloads on ROCm/HIP, CUDA, OpenCL, and heterogeneous accelerators
  • Design and own Python-based automation frameworks for validation, benchmarking, and reporting
  • Drive improvements in validation scalability, efficiency, and coverage
  • Collaborate with compiler, runtime, driver, and hardware teams
  • Provide technical leadership and mentorship to senior and junior engineers
  • Communicate validation status, performance metrics, and quality risks to stakeholders

Qualifications

  • 8–12 years of experience in AI/ML software validation or performance engineering
  • Strong expertise in Python scripting and test automation
  • Strong ML fundamentals including deep learning and LLMs
  • Hands-on experience with ROCm validation, performance profiling, and optimization
  • Experience with HIP, CUDA, OpenCL, and TensorFlow/PyTorch integrations
  • Proven experience validating end-to-end AI pipelines
  • Strong Linux expertise (Ubuntu, Yocto)
  • Performance-driven validation mindset with focus on production readiness
  • Ability to lead initiatives independently with strong ownership
  • Strong analytical and problem-solving skills with clear written and verbal communication, and ability to collaborate with global, cross-functional teams
  • Bachelor’s or Master’s degree in AI/ML, Robotics, Electronics, Computer Science, or related field

Skills

Required:Python scripting and test automationDeep learning fundamentalsROCm validation, profiling, and optimizationHIP, CUDA, OpenCLEnd-to-end AI pipeline validationLinux (Ubuntu, Yocto)

Benefits

  • AMD benefits at a glance
Last updated Oct 22, 2026. Interested? Apply on the company site.
AMD logo

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
Company profileWebsite

More open roles at AMD