Software Engineer, AI Infrastructure & Testing
Core
Design, maintain, and evolve custom ML model benchmarking and validation infrastructure to measure correctness, performance, and regressions across compiler releases.
Role type
Senior IC software engineer (AI infrastructure & testing)
Builds
Scalable test pipelines, benchmarks, tooling, and CI/CD infrastructure for an ML compiler platform
Domain
AI compilers, software testing, systems programming
Deliverable
production ML models | infrastructure
Required skills
Python, C/C++, software engineering principles, data structures, design patterns, algorithms
Preferred skills
Quality Assurance or Test Engineer experience, compiler experience (MLIR, LLVM), compute frameworks (Vulkan, Cuda, Metal, OpenCL), AI algorithms knowledge
Technologies
Python, C, C++, MLIR, LLVM, Vulkan, Cuda, Metal, OpenCL
Responsibilities
Design and maintain ML model benchmarking and validation infrastructure; Validate compiled models within third-party software stacks; Collaborate with compiler, runtime, and ML engineers to diagnose failures; Build scalable test pipelines covering diverse models and hardware; Contribute test cases and tooling to improve platform reliability; Own the evolution of the build system and CI/CD infrastructure
Seniority
Senior, hands-on IC
