AI ML Test Engineer

We are seeking a highly skilled AI/ML Validation Engineer with a strong foundation in machine learning, deep learning, and system-level validation. The ideal candidate will have hands-on experience with ML frameworks, profiling tools, and AI compute stacks, and will play a key role in validating end-to-end AI pipelines and ensuring software quality across diverse platforms.

Responsibilities

• Design and execute validation plans for AI/ML compute stacks (HIP, CUDA, OpenCL, OpenVINO, ONNX Runtime, TensorFlow, PyTorch).
• Validate end-to-end AI pipelines including model conversion, inference runtimes, compilers/toolchains, kernel execution, and memory transfer.
• Profile ML workloads and optimize performance across platforms.
• Collaborate with global teams to ensure high-quality deliverables.
• Apply software development lifecycle practices to validation workflows.
• Document defects, validation results, and improvement recommendations.

Primary Skills

• Strong knowledge of ML fundamentals, deep learning, LLMs, and recommender systems.
• Proficiency in Python programming.
• Experience with PyTorch, TensorFlow, ONNX Runtime.
• Familiarity with Ubuntu/Yocto Linux environments.
• Expertise in profiling tools and performance analysis.
• Strong understanding of software QA methodologies.

Secondary Skills

• Experience with compilers/toolchains like TVM, Vitis AI, XDNA, XLA.
• Exposure to production-grade software validation workflows.
• Excellent problem-solving and communication skills.

Educational Qualification

• Bachelor’s or Master’s degree in Computer Science, Electrical Engineering, or related field.