d-Matrix

Software Machine Learning Test Engineer - Sr Engineer

d-Matrix
Apply
6 months ago
Bengaluru, IndiaSenior
H1B Sponsor

Base Salary

$155k - $234k/yr

Responsibilities

  • Develop test automation for d-Matrix software components and products.
  • Maintain and improve test infrastructure for next-generation AI hardware.
  • Build and scale test frameworks for software stack and hardware product qualification.
  • Use GitLab infrastructure and Pytest to perform component and end-to-end qualification.
  • Develop automation for large language model accuracy, benchmarking, and robustness testing.
  • Collaborate with ML kernel, compiler, runtime, hardware architecture, and machine learning research teams during qualification.
  • Contribute to new ML and AI technologies.

Requirements

  • Bachelor's or master's degree in Computer Science, Electrical Engineering, or a related field.
  • At least 7 years of industry experience.
  • Knowledge of large language models and machine learning fundamentals.
  • Knowledge of Python programming, Python data structures, and the Linux environment.
  • Experience with machine learning models and their layers is desired.
  • Experience with deep learning, TensorFlow, Transformers, and PyTorch testing is desired.
  • Knowledge of GitLab CI/CD infrastructure is desired.
  • Prior startup or small-team experience is desired.

Benefits

  • Hybrid work arrangement with three days onsite per week at the Bangalore, India headquarters
  • Equal opportunity workplace committed to an inclusive environment

Tech Stack

LinuxpytestPythonPyTorchTensorFlow

Categories

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.