AMD logo

Lead AI/ML Software Use Cases Validation Engineer

AMD · Posted Oct 10

High-performance processors, graphics cards, data center solutions, and AI accelerators

Bangalore, IndiaFull-timeOnsiteLead/Staff8–12 years₹45.0L–₹75.0L yearly51 applicants
SemiconductorsHardwareArtificial IntelligenceCloud ComputingGamingPublic Company
Be an early applicantFull time

Get a personal compatibility score

Add a resume for personal matches

About the role

AMD's PAVS team validates robotics platforms, variants, and customer use cases across software, hardware, and integrated systems, ensuring safety, reliability, and performance in simulation, SIL/HIL, and real-world deployments. The role leads validation, benchmarking, and performance optimization of AI/ML compute stacks and Physical AI SDKs on Linux platforms, working closely with architecture, compiler, runtime, and hardware teams. It also covers end-to-end AI pipeline validation, defect lifecycle ownership, Python-based automation frameworks, and technical mentorship.

What you will do

  • 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
  • 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 performance coverage
  • Collaborate closely 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

Skills used in this role

PythonPyTorchTensorFlowONNXONNX RuntimeROCmHIPCUDAOpenCLLinuxUbuntuYoctoDeep LearningLLMsTest AutomationPerformance ProfilingBenchmarkingProblem SolvingCommunicationLeadershipMentorship

What the employer is looking for

  • 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 the ability to collaborate effectively with global, cross‑functional teams.
  • Bachelor’s or Master’s degree in AI/ML, Robotics, Electronics, Computer Science, or a related field.

Benefits and support

  • Compensation and benefits are detailed in the job posting

About AMD

AMD is a global semiconductor company that designs high-performance central processing units (CPUs), graphics processing units (GPUs), and adaptive processors for data centers, personal computers, gaming consoles, and embedded systems. The company specializes in advanced computing solutions and AI accelerators, positioning itself as a major leader in modern high-performance and artificial intelligence hardware. Through strategic developments and acquisitions, AMD continues to expand its ecosystem across cloud infrastructure, enterprise computing, and physical AI.

Industry
Semiconductors
Company size
31000 employees
Founded
1969-05-01
Location
Santa Clara, California, USA
Funding stage
Public Company

Funding

Public Company

  • 2007-11-16Post Ipo

Leadership

DL
Dr. Lisa Su

Chair and Chief Executive Officer

JH
Jean Hu

Executive Vice President, Chief Financial Officer and Treasurer

MP
Mark Papermaster

Executive Vice President and Chief Technology Officer

PG
Philip Guido

Executive Vice President and Chief Commercial Officer

FN
Forrest Norrod

Executive Vice President and General Manager, Data Center Solutions Business Group