Software Machine Learning Test Engineer - Staff

📋 Description

• • Join d-Matrix Corporation, a pioneering company at the forefront of generative AI innovation, and play a pivotal role in shaping the future of AI hardware and software.
• • As a Staff Software Machine Learning Test Engineer, you will be instrumental in productizing the software stack for d-Matrix's cutting-edge AI compute engine, ensuring the delivery of highly qualified software and hardware products.
• • You will be a key member of the Software Quality Assurance (QA) team, focusing on the development and maintenance of robust test automation for d-Matrix components and products.
• • Your responsibilities will include enhancing and scaling the test infrastructure to support next-generation AI hardware, ensuring its reliability and performance.
• • Leverage your logical thinking and engineering prowess to build and scale a comprehensive test framework that underpins the quality of our advanced AI solutions.
• • Actively participate in the d-Matrix GitLab CI/CD infrastructure, utilizing the Pytest framework to achieve rigorous component and end-to-end qualification of our software stack.
• • Develop and implement sophisticated testing methodologies for large language models (LLMs), focusing on accuracy, performance benchmarking, and robustness.
• • Design and execute automation scripts to validate LLM performance and identify potential issues, contributing directly to the productization of our AI compute engine.
• • Collaborate closely with global Machine Learning (ML) core kernel teams, compiler teams, runtime teams, and hardware architecture experts throughout the qualification process.
• • Engage with machine learning model researchers to understand their requirements and ensure the software stack meets the demands of advanced AI workloads.
• • Contribute to the exploration and integration of new ML/AI technologies, staying at the cutting edge of the field.
• • This role offers a unique opportunity to work on challenging problems at the intersection of software, hardware, and artificial intelligence, contributing to a culture of respect, collaboration, and direct communication.
• • Embrace a culture that values humility, inclusivity, and diverse perspectives, fostering an environment where innovative solutions are born from collective effort.
• • Be part of a team driven by execution, passionate about tackling complex challenges, and dedicated to shaping the endless possibilities of AI.
• • This is a hybrid role, requiring you to work onsite at our Bangalore, India headquarters three days per week, fostering in-person collaboration and team synergy.
• • You will be empowered to find your 'playground' – a space where your skills and creativity can flourish, and where you can contribute meaningfully to groundbreaking AI advancements.
• • Your work will directly impact the quality and performance of d-Matrix's AI compute engine, a critical component in the rapidly evolving landscape of artificial intelligence.
• • You will gain exposure to a wide array of AI technologies and methodologies, from foundational ML concepts to the latest advancements in LLMs and deep learning frameworks.
• • The role demands a proactive approach to identifying and mitigating risks in the software and hardware development lifecycle, ensuring a high level of product quality.
• • You will have the opportunity to influence the direction of testing strategies and infrastructure development within the organization.
• • Contribute to building a scalable and maintainable test automation suite that can adapt to future product iterations and technological advancements.
• • This position is ideal for an experienced engineer who thrives in a fast-paced, innovative environment and is eager to make a significant impact in the AI domain.