d-Matrix · 10 hours ago
Software ML Test Engineer, Senior Staff
Santa Clara, CA
Full-time
Hybrid
Senior Level, Lead/Staff
5+ years expd-Matrix is focused on unleashing the potential of generative AI to power the transformation of technology. They are seeking a Senior Staff Software ML Test Engineer to lead junior team members in test planning, automation, and execution while collaborating with various teams to ensure thorough testing of the AI compute engine.
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 with software development teams to design, automate, and execute software and system use cases applicable to features and functionalities
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, software development team, and systems team on test coverage and test results
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 and apply them to testing
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
2025-12-22
2025-12-17
Crunchbase News
2025-12-10
Company data provided by crunchbase