about 6 hours ago
Base Salary
$170k - $200k/yr
Responsibilities
- Design, build, and maintain scalable automated test frameworks for kernel validation.
- Develop correctness and performance regression suites for ML operators.
- Implement testing pipelines integrated into CI/CD workflows.
- Build tooling to automate comparison of kernel outputs against reference implementations.
- Develop test coverage for software kernels targeting specialized hardware.
- Validate numerical precision and edge cases for ML operators.
- Partner with hardware teams to validate hardware-software co-design assumptions.
- Act as the primary liaison between the Kernels engineering team and the QA organization.
- Participate in QA planning ceremonies and surface quality risks early.
- Set quality standards and best practices for the Kernels team.
Requirements
- BSc in Computer Engineering, Computer Science, Math, Physics, or related field with 6+ years of experience.
- Strong understanding of computer architecture and machine learning fundamentals.
- Proficient in C/C++ and Python, with experience writing production-quality test code.
- Experience writing automated tests for algorithms targeting specialized hardware.
- Familiarity with ML operators commonly used in production workloads.
- Experience building or working within CI/CD pipelines and automated regression systems.
- Track record of owning quality outcomes and defining strategy and coverage.
- Strong cross-functional communication skills.