d-Matrix

Staff Software Engineer in Test, AI Kernels

Santa Clara · Staff+

Sponsorship not specifiedDetected 6 days ago

PythonC++Data StructuresAlgorithmsCI/CDLinuxMachine LearningTensorFlowPyTorchLLMsSignal ProcessingLeadershipCommunication

About the role

You will serve as the primary quality interface between the Kernels team and the broader QA organization, driving alignment on test strategy, coverage, and release readiness.
This is a high-impact, deeply technical role that requires both strong software engineering skills and a rigorous quality mindset.

Design, build, and maintain scalable automated test frameworks for kernel validation across simulation, emulation, and silicon targets.
Develop correctness and performance regression suites covering ML operators (GEMMs, convolutions, BLAS, SIMD ops, softmax, layer norm, pooling, etc.).
Implement testing pipelines integrated into CI/CD workflows, enabling rapid feedback on kernel changes.
Build tooling to automate comparison of kernel outputs against reference implementations (e.g., CPU-based golden references, PyTorch/TensorFlow baselines).
Validate numerical precision, edge cases, and boundary conditions for ML operators across data types and hardware configurations.
Partner with hardware teams (mixed signal, DSP, CPU) to validate hardware-software co-design assumptions and catch integration issues early.
Drive validation of compiler-generated code paths (MLIR, LLVM, TVM, etc.) through structured test methodologies.
Contribute to and help maintain shared QA infrastructure, test standards, and reporting dashboards used across engineering.
Drive root cause analysis for test failures and escaped defects; work with Kernels engineers to close gaps.

BSc in Computer Engineering, Computer Science, Math, Physics, or a related field with 6+ years of industry experience
or MS with 4+ years of experience
or PhD with 2+ years of experience, with at least 2 years focused on test engineering, SDET, or quality engineering roles.
experience writing production-quality test code in these languages in Linux environments.
Experience writing automated tests for algorithms targeting specialized hardware such as GPUs, DSPs, FPGAs, or AI accelerators (e.g., using CUDA or equivalent).
Familiarity with ML operators commonly used in production workloads: GEMMs, convolutions, BLAS, SIMD operations, softmax, layer normalization, pooling, and similar.

Experience embedded in a hardware-adjacent software team (AI accelerator company, cloud compute, or similar) rather than a purely software QA role.
Hands-on experience with ML frameworks such as PyTorch or TensorFlow, including the ability to write reference implementations for operator validation.
Familiarity with ML compiler toolchains (MLIR, LLVM, TVM, Glow, etc.) and how to test compiler-generated artifacts.
Experience with embedded SIMD vector processors such as Tensilica.
Prior startup or small-team experience
comfort operating with high ownership and limited process overhead.
Experience defining and implementing formal test plans and coordinating release qualification with a QA organization.
You will have a direct impact on the correctness and quality of AI compute software that ships in next-generation hardware products.

Develop test coverage for software kernels targeting specialized hardware including AI accelerators, DSPs, FPGAs, and SIMD vector processors (e.g., Tensilica).
Act as the primary liaison between the Kernels engineering team and the QA organization, aligning on test plans, coverage criteria, and release qualification gates.

At d-Matrix, we are focused on unleashing the potential of generative AI to power the transformation of technology.
We are at the forefront of software and hardware innovation, pushing the boundaries of what is possible.
Our culture is one of respect and collaboration.
We value humility and believe in direct communication.
Our team is inclusive, and our differing perspectives allow for better solutions.
We are seeking individuals passionate about tackling challenges and are driven by execution.
Ready to come find your playground?
Together, we can help shape the endless possibilities of AI.
You will work alongside compiler engineers, ML software engineers, and hardware architects while partnering closely with QA leads to define standards and share best practices across the company.

Employment Policy
d-Matrix is proud to be an equal opportunity workplace and affirmative action employer. We're committed to fostering an inclusive environment where everyone feels welcomed and empowered to do their best work. We hire the best talent for our teams, regardless of race, religion, color, age, disability, sex, gender identity, sexual orientation, ancestry, genetic information, marital status, national origin, political affiliation, or veteran status. Our focus is on hiring teammates with humble expertise, kindness, dedication and a willingness to embrace challenges and learn together every day.
d-Matrix does not accept resumes or candidate submissions from external agencies. We appreciate the interest and effort of recruitment firms, but we kindly request that individual interested in opportunities with d-Matrix apply directly through our official channels. This approach allows us to streamline our hiring processes and maintain a consistent and fair evaluation of al applicants. Thank you for your understanding and cooperation.
d-Matrix is proud to be an equal opportunity workplace and affirmative action employer.

This listing is sourced directly from d-Matrix's careers page and normalized into a canonical job model.