d-Matrix · 1 day ago
Software ML Test Engineer, Senior Staff
Santa Clara
Full-time
Hybrid
Senior Level, Lead/Staff
$160K/yr - $250K/yr
5+ years expd-Matrix is focused on unleashing the potential of generative AI to transform technology. The Senior Staff Software ML Test Engineer will lead junior team members in test planning, automation, and execution to develop test suites and infrastructures for the d-Matrix AI compute engine, collaborating with various teams on software testing and debugging.
AI InfrastructureArtificial Intelligence (AI)Cloud InfrastructureData CenterSemiconductor
Responsibilities
Lead junior team members with test planning, test automation, and test execution to help productize the d-Matrix AI compute engine
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 both software and system use cases that are applicable to the features and functionalities as well as the deployment scenarios for the next-gen AI chip software stack
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 the pytest framework along with Jenkins and GitLab
Collaborate with the product team, the software development team, and the systems team on the test coverage and the test results
Fully engaged with the test case implementation and the software bug triaging and debugging
Qualification
Required
MS or PhD preferred in Computer Science, Electrical Engineering, or related degree with 5+ 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
Preferred
Prior startup, small team experience
Experience with Machine Learning models and their layers
Experience testing and automating at-scale ML kernels, MLIR, LLVM, Matmuls, and Numpy workloads
Fluency with Gitlab CI/CD infrastructure
Company
d-Matrix
D-Matrix is a platform that enables data centers to handle large-scale generative AI inference with high throughput and low latency.
51-200 employees
H1B Sponsorship
d-Matrix has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (20)
2024 (15)
2023 (8)
2022 (7)
Funding
Current Stage
Growth StageTotal Funding
$429MKey Investors
Temasek HoldingsTSVC
2025-11-12Series C· $275M
2023-09-06Series B· $110M
2022-04-20Series A· $44M
Leadership Team
Peter Buckingham
Senior Vice President, Software Engineering
Recent News
2026-01-22
2025-12-22
2025-12-17
Company data provided by crunchbase