Sr. Staff, Software Machine Learning Test Engineer
Core
Developing test automation and infrastructure to qualify software stacks and hardware products for a generative AI compute engine.
Role type
Sr. Staff Software Machine Learning Test Engineer
Builds
Test automation frameworks and qualification pipelines for AI compute engine components and products
Domain
Generative AI hardware and software stack
Deliverable
production ML models
Required skills
Python programming, Linux environment, GitLab CI/CD, Pytest, large language model fundamentals, machine learning fundamentals, deep learning testing, transformer testing, PyTorch testing, Tensorflow testing
Preferred skills
Startup experience, small team experience, ML model layer knowledge
Technologies
GitLab CI/CD, Pytest, Tensorflow, PyTorch, Transformers
Responsibilities
Develop test automation for d-Matrix components and products, maintain and improve test infrastructure for next-generation AI hardware, leverage GitLab CI/CD and Pytest for component and end-to-end qualification, develop automation for LLM accuracy benchmarking and robustness testing, collaborate with ML core kernel, compiler, runtime, hardware architecture, and model research teams
Seniority
Sr. Staff, hands-on IC
