Software Engineer, AI Infrastructure & Testing
Core
Designing and maintaining custom ML model benchmarking, validation, and CI/CD infrastructure for an AI compiler platform running on heterogeneous edge hardware.
Role type
Senior IC software engineer (AI infrastructure & testing)
Builds
Scalable test pipelines, benchmarking tools, and CI/CD systems for an ML compiler stack
Domain
Edge AI, compilers, heterogeneous hardware (CPUs, GPUs, NPUs)
Deliverable
production ML models | infrastructure
Required skills
Python, C/C++, software engineering principles, data structures, algorithms, build systems, CI/CD
Preferred skills
Quality Assurance/Test Engineering, compilers (MLIR/LLVM), compute frameworks (Vulkan/Cuda/Metal/OpenCL), AI algorithms, hardware architecture knowledge
Technologies
Python, C, C++, MLIR, LLVM, Vulkan, Cuda, Metal, OpenCL
Responsibilities
Design and evolve custom ML model benchmarking and validation infrastructure; Validate compiled models within third-party stacks to uncover integration issues; Collaborate with compiler and runtime 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
