Lead AI/ML Validation Engineer

Advanced Micro Devices, Inc
Posted on
Advanced Micro Devices, Inc logo

Experience
Up to 2 yrs
Job Location
Bengaluru, India
Vacancy
1
Designation
Lead Machine Learning Engineer
Job Type
Not specified

Job Description

Role Overview
We are seeking an experienced SMTS Software Quality Engineer to join the HIP Runtime Quality Engineering team within the ROCm ecosystem. In this role, you will focus on defect triage, root cause analysis, and scalable test automation for the HIP runtime, ensuring robustness, portability, and performance across AMD and NVIDIA GPU backends.
The role sits at the intersection of runtime systems, GPU APIs, and CI-driven quality assurance , working closely with HIP runtime developers, compiler teams, and system software engineers to improve test coverage and accelerate issue resolution across a fast-evolving open source stack.
HIP is a C++ runtime API and kernel language designed for CUDA portability and heterogeneous GPU programming in the ROCm platform.
Key Responsibilities
  • Own end-to-end triage of HIP runtime failures observed in CI, nightly, and release testing pipelines.
  • Analyze failures across HIP runtime APIs , including memory management, streams, events, kernels, graphs, and device management.
  • Perform log analysis, repro reduction, and backtrace debugging across Linux and Windows environments.
  • Correlate failures across:
    • ROCr system runtime
    • HIP runtime
    • Driver / firmware interfaces
    • Compiler (hipcc / clang) interactions
  • File high quality bug reports with clear reproduction steps, root cause hypotheses, and impact assessment .
Test Automation Framework Development
  • Design, enhance, and maintain automated test frameworks for HIP runtime validation using existing HIP test infrastructure (Catch2-based and custom harnesses).
  • Expand functional, negative, stress, and regression coverage for:
    • HIP Runtime APIs
    • Graph APIs
    • Memory copy variants (1D/2D/3D, async/sync)
    • Multi-GPU and peer-to-peer use cases
  • Integrate tests into CI/CD pipelines with actionable failure reporting.
  • Develop Python, C++, and shell-based automation for test execution, result aggregation, and trend analysis.
  • Drive test flakiness reduction and improve signal-to-noise ratio in large-scale CI.
Cross-Team Collaboration
  • Partner with HIP runtime developers to:
    • Review test gaps for new features and APIs
    • Validate bug fixes and performance-sensitive code paths
  • Collaborate with:
    • Compiler and language runtime teams
    • ROCm system runtime teams
    • Release engineering and CI infrastructure teams
  • Contribute test fixes and enhancements upstream in open-source HIP/ROCm repositories .
Quality Metrics Continuous Improvement
  • Define and track quality metrics such as failure escape rate, test coverage, and mean-time-to-resolution MTTR.
  • Identify systemic issues and propose data-driven quality improvements .
  • Participate in release readiness assessments for HIP runtime drops.
Required Qualifications Technical Skills
  • Strong experience in software quality engineering, test automation, or SDET roles .
  • Solid proficiency in C++ with debugging experience (gdb/lldb).
  • Working knowledge of HIP or CUDA runtime concepts , including kernels, streams, memory models, and device execution.
  • Experience with Linux-based development and debugging .
  • Familiarity with CI systems (Jenkins, GitHub Actions, GitLab CI, or similar).
  • Experience writing and maintaining automated tests for low-level systems or runtime libraries.
Debugging Analysis Skills
  • Ability to reason about runtime abstractions , concurrency, and asynchronous execution.
  • Strong skills in failure reproduction, minimization, and log-driven debugging .
  • Experience triaging issues across hardware software boundaries is a plus.
Preferred Qualifications
  • Experience with GPU programming models (HIP, CUDA, OpenCL).
  • Familiarity with ROCm stack components (HIP, ROCr, compilers, drivers).
  • Experience testing graph APIs, async execution models, or multi-GPU systems .
  • Prior contributions to open-source system software or test frameworks .
  • Knowledge of Python-based test orchestration and reporting tools .
  • Knowledge and application of AI tools in validation and testing.
Disclaimer: This job posting has been aggregated from external source. Role details, content, and availability are subject to change. Applicants are advised to confirm the latest information directly on the company website before applying.

No Referrers Available

There are currently no referrers available for this job. You can still apply, will let you know once there is any referrer available.