Senior software engineer (machine learning quality assurance)

Requirements

• Applicants for this role should have strong experience designing, developing, and maintaining high-quality software systems
• The ideal candidate is an experienced software engineer who values code quality, testability, and long-term maintainability, and enjoys building systems that other engineers rely on
• This person will be comfortable working across large codebases, contributing to CI/CD infrastructure, and shaping technical direction through thoughtful design and mentoring in a technically demanding environment spanning ML frameworks, infrastructure, and AI accelerator hardware
• Experience in production-quality software engineering roles
• Strong software design and architecture skills, with experience working on large or complex systems
• Strong proficiency in Python, including experience building and maintaining production codebases
• Solid experience with CI/CD systems and automated testing (preferably GitHub-based workflows)
• Experience working in Linux environments
• Familiarity with C or C++, with the ability to read, debug, and reason about low-level code when needed
• Proven ability to mentor junior engineers and influence engineering practices within a team
• Strong problem-solving skills and a proactive, self-directed approach to work
• Bachelor/Master's/PhD or equivalent experience in Computer Science, Maths, Machine Learning, Data Science, or related field
• (Desirable) Exposure to machine learning frameworks such as PyTorch, JAX, Triton,TensorFlow
• (Desirable) Experience with distributed workload management systems such asKubernetes, VLLM, Keras or MLOpspipelines
• (Desirable) Experience working with hardware simulators or emulators (e.g. QEMU)
• (Desirable) Experience developing for or working with FPGA-based systems
• (Desirable) Experience with people management or mentoring

What the job involves

• The role focuses on testing and validating a complex machine learning software stack, with particular emphasis on software architecture, automation, and engineering best practices
• The ML QA team is composed of highly skilled software engineers with a strong focus on automation, software quality, and data-driven validation. The team works closely with industry-standard machine learning frameworks and models, contributing to upstream open-source projects and collaborating across the wider software organization
• Operating in a fast-paced environment, the team plays a critical role in ensuring reliability, performance, and maintainability across the ML software stack, helping to deliver robust and high-quality products to customers
• Design, implement, and maintain robust test infrastructure and automation for a complex ML software stack
• Architect and evolve test frameworks and tooling with a focus on scalability, maintainability, and developer experience
• Build and maintain CI/CD pipelines targeting simulators, emulators (e.g. QEMU), and physical hardware
• Create representativeML workloadsand gain insights from their execution. (Numerical accuracy, performance analysis and benchmarking)
• Work closely with all Software development teams, supporting a culture of quality, security and maintainability
• Review code and designs, setting a high bar for software engineering best practices
• Mentor and support junior engineers, helping raise the overall technical capability of the team
• Evaluate existing test strategies and infrastructure, identifying gaps and driving improvements aligned with team and organizational goals