Software Machine Learning Test Engineer

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.