About this role

Help productize the SW stack for the d-Matrix AI compute engine
Responsible for test automation development for d-Matrix components and products
Maintain and improve test infrastructure for next-generation AI hardware
Build and scale the test framework to deliver highly qualified software stacks and hardware products
Leverage the d-Matrix GitLab CI/CD infrastructure and Pytest framework for testing
Develop large language model accuracy, benchmarking, and robustness testing using self-developed automation
Work with global ML core kernel teams, compiler teams, runtime teams, hardware architecture experts, and ML model researchers during the qualification process

Bachelor's or master's degree in Computer Science, Electrical Engineering, or related degree with 7+ Years of Industry Experience
Good knowledge of large language models and machine learning fundamentals
Good knowledge of Python programming, Python Data Structures, and Linux environment
Self-motivated team player
Prior startup, small team experience
Experience with Machine Learning models and their layers
Experience with Deep Learning, Tensorflow, Transformers, and PyTorch testing
Knowledge of Gitlab CI/CD infrastructure

Staff Software Machine Learning Test Engineer

Key skills