Senior Ai/ml Software Use Cases Validation Engineer

AMD AMD · Semiconductors · Bangalore, India · Engineering

Senior AI/ML Software Use Cases Validation Engineer at AMD in Bangalore, India. Focuses on end-to-end AI pipeline validation, benchmarking, and performance optimization across Linux platforms, including model training, conversion, optimization, kernel execution, and inference accuracy. Requires strong Python scripting, automation, ML fundamentals (deep learning, LLMs), experience with AI frameworks (PyTorch, TensorFlow, ONNX Runtime), and accelerators (ROCm/HIP, CUDA, OpenCL).

What you'd actually do

  1. Validation of AI/ML compute stacks on Ubuntu and Yocto
  2. Define and execute validation plans covering functional, performance, stress, regression, and scalability testing
  3. Own defect lifecycle including triage, root cause analysis, and closure
  4. Validate end‑to‑end AI pipelines: Model training, conversion, and optimization (e.g., PyTorch → ONNX)
  5. Define and run AI benchmarking and profiling for training and inference workloads

Skills

Required

  • Python scripting and automation
  • ML fundamentals
  • deep learning
  • LLMs
  • HIP
  • CUDA
  • OpenCL
  • TensorFlow/PyTorch integrations
  • ROCm validation
  • end-to-end AI pipeline validation
  • Linux expertise (Ubuntu, Yocto)
  • analytical and problem-solving skills
  • written and verbal communication
  • collaboration

Nice to have

  • Benchmarking on CPU, GPU, and AI accelerators
  • Performance tuning of large models (memory optimization, mixed precision)

What the JD emphasized

  • end-to-end AI pipeline validation
  • benchmarking
  • performance optimization
  • model training
  • conversion
  • optimization
  • kernel execution
  • inference accuracy
  • AI benchmarking
  • profiling
  • training
  • inference workloads
  • ROCm validation

Other signals

  • validation of AI/ML compute stacks
  • end-to-end AI pipeline validation
  • benchmarking and performance optimization
  • AI frameworks and runtimes validation