Principal Software Engineer – ML Test

d-Matrix

Principal Software ML Test Engineer leading test automation for d-Matrix's AI compute engine. Collaborating with development teams to enhance testing processes.

Posted 4/1/2026full-timeSanta Clara • California • 🇺🇸 United StatesLead💰 $180,000 - $300,000 per yearWebsite

Tech Stack

Tools & technologies

JenkinsLinuxNumpyPythonPyTorchTensorflow

About the role

Key responsibilities & impact

Lead junior team members with test planning, test automation, and test execution
Responsible for the development of the test suites, the test cases, and the test infrastructures
Work side by side with the software development teams to design, automate, and execute software and system use cases
Implement Python programs to test workloads and ML models within the Pytest framework
Implement and improve the d-Matrix end-to-end test suite along with the regression test suite in CI/CD infrastructure using pytest, Jenkins, and GitLab
Collaborate with the product team, software development team, and systems team on the test coverage and test results
Fully engage with test case implementation and software bug triaging and debugging
Fluent in advanced Python programming with the ability to write data structures and algorithms

Requirements

What you’ll need

MS or PhD preferred in Computer Science, Electrical Engineering, or related degree with 10+ Years of Industry Experience
Strong grasp of large language models and machine learning fundamentals
Experience with Deep Learning, Tensorflow, Transformers, and PyTorch testing
Proficient in Python development, Python Data Structures, and Linux environments
Self-motivated team player with a strong sense of ownership and leadership
Prior startup, small team experience (Desired)
Experience with Machine Learning models and their layers (Desired)
Experience testing and automating at-scale ML kernels, MLIR, LLVM, Matmuls, and Numpy workloads (Desired)
Fluency with Gitlab CI/CD infrastructure (Desired)

Benefits

Comp & perks

Offers Equity
Offers Bonus
Health insurance
Retirement plans
Paid time off
Flexible work arrangements
Professional development

ATS Keywords

✓ Tailor your resume

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools

Pythontest automationtest planningtest executiondata structuresalgorithmsDeep LearningTensorflowPyTorchML models

Soft Skills

leadershipself-motivatedteam playerownership

Certifications

MS in Computer SciencePhD in Computer ScienceMS in Electrical EngineeringPhD in Electrical Engineering