AI/ML Compiler & Runtime Software Engineer
Core
Architecting and developing AI/ML compiler and runtime software to enable efficient execution of AI models on edge devices and RISC-V based NPU/SoC platforms.
Role type
Sr Staff Engineer, AI/ML Compiler & Runtime Software
Builds
IREE-based compiler flows, custom MLIR dialects, runtime integration, and deployment paths for edge AI workloads on silicon platforms.
Domain
Semiconductor / Edge AI / Compiler Infrastructure
Deliverable
production ML models
Required skills
IREE, LLVM, MLIR, C/C++, Python, neural network optimization, quantization, operator fusion, memory planning, hardware-software co-design, Linux development, cross-compilation, technical leadership
Preferred skills
RISC-V, FPGA prototyping, LLM inference stacks, CI/CD pipelines, silicon bring-up
Technologies
IREE, LLVM, MLIR, PyTorch, ONNX, TensorFlow Lite, torch-mlir, TOSA, Linalg, Jenkins, Git, CMake, Bazel, Jira
Responsibilities
Architect AI/ML compiler and runtime software for RISC-V based IP; Develop IREE-based compiler flows and custom MLIR dialects; Optimize neural network workloads for edge deployment; Enable efficient execution across CPU, vector, matrix, and NPU paths; Collaborate with hardware and firmware teams to bring up AI workloads; Define software architecture and technical direction for AI SDK components; Build test infrastructure and validation flows; Provide technical leadership to engineering teams.
Seniority
Sr Staff, hands-on IC with technical leadership
