Search by job, company or skills

Software Machine Learning Test Engineer

Early Applicant
  • Posted 22 days ago
  • Be among the first 10 applicants

Job Description

Role Overview

You will be part of a team responsible for bringing a software stack to production for next-generation AI compute platforms. As a member of the Software QA team, you will design and build automated test frameworks for AI hardware and software components, and continuously enhance the test and qualification infrastructure to support evolving products.

This role involves leveraging CI/CD pipelines and Python-based test frameworks to drive both component-level and end-to-end validation. You will develop automation to support large language model accuracy, benchmarking, and robustness testing, ensuring software and hardware stacks meet high quality and performance standards.

You will collaborate closely with machine learning kernel teams, compiler and runtime engineers, hardware architects, and ML researchers across global teams throughout the qualification lifecycle. There will also be opportunities to contribute to emerging ML and AI technologies as part of the broader engineering effort.

Qualifications

  • Bachelor's or Master's degree in Computer Science, Electrical Engineering, or a related field, with 7+ years of industry experience.
  • Solid understanding of machine learning fundamentals and large language models.
  • Strong proficiency in Python, including data structures, and experience working in a Linux environment.
  • Self-driven team player with strong collaboration skills.

Preferred Experience

  • Experience working in startups or small, fast-paced teams.
  • Hands-on exposure to machine learning models and neural network layers.
  • Experience testing deep learning frameworks such as TensorFlow, PyTorch, and transformer-based models.
  • Familiarity with CI/CD systems, particularly GitLab-based pipelines.

More Info

Job Type:
Industry:
Employment Type:

About Company

Job ID: 150551489

Similar Jobs

Bengaluru, India

Skills:

TensorflowMachine LearningPytorchLinuxPythonPytestDeep LearningTransformers