d-Matrix

Principal Software ML Test Engineer

d-Matrix
Apply
5 months ago
Santa Clara, CA, USAStaff+
H1B Sponsor

Base Salary

$180k - $300k/yr

Responsibilities

  • Lead junior team members in test planning, test automation, and test execution.
  • Develop test suites, test cases, and test infrastructure for the d-Matrix AI compute engine.
  • Design, automate, and execute software and system use-case tests across features, functionality, and deployment scenarios.
  • Implement Python programs to test workloads and machine-learning models within the Pytest framework.
  • Build and improve end-to-end and regression test suites using Jenkins and GitLab infrastructure.
  • Collaborate with product, software development, and systems teams on test coverage and results.
  • Implement test cases and perform software bug triage and debugging.

Requirements

  • 10+ years of industry experience.
  • MS or PhD in Computer Science, Electrical Engineering, or a related field preferred.
  • Strong understanding of large language models and machine-learning fundamentals.
  • Experience testing Deep Learning, TensorFlow, Transformers, and PyTorch.
  • Proficiency in Python development, Python data structures, algorithms, and Linux environments.
  • Experience with machine-learning models and their layers is desired.
  • Experience testing and automating at-scale ML kernels, MLIR, LLVM, Matmuls, and NumPy workloads is desired.
  • Fluency with GitLab CI/CD infrastructure is desired.
  • Prior startup or small-team experience is desired.

Benefits

  • Hybrid work arrangement with 3 days per week onsite at the Santa Clara, California headquarters.

Tech Stack

JenkinsLinuxNumPypytestPythonPyTorchTensorFlow
d-Matrix

About d-Matrix

201-500 employees

d-Matrix is an AI infrastructure company building the next generation of inference computing for the era of generative and agentic AI. Founded in 2019, d-Matrix is rethinking AI inference from the ground up with a full-stack approach spanning silicon, systems, networking, and software. Its flagship products, including the Corsair™ inference platform and JetStream™ inference fabric, are purpose-built to deliver high-performance, low-latency, and energy-efficient AI inference at datacenter scale. The company has raised nearly $500 million from a global syndicate of leading venture capital firms, sovereign wealth funds, and strategic investors, including Playground Global, Bullhound Capital, M12 (Microsoft’s Venture Fund), SK hynix, Temasek, Qatar Investment Authority, and Singapore’s EDBI. The company’s most recent financing valued d-Matrix at approximately $2 billion. As AI shifts from training to inference, the demands on AI infrastructure are changing. d-Matrix is building the infrastructure required to power the next generation of real-time AI at scale.