AI/ML Compiler & Runtime Software Engineer

About the role

We are seeking a highly skilled Sr Staff Engineer in AI/ML compiler and runtime software to join our Platform Software and AI SDK team. The team is building the foundational software stack to enable Physical AI workloads on next-generation RISC-V IP and SoC platforms. This role sits at the intersection of AI compiler technology, edge AI deployment, runtime systems, and hardware acceleration.

What you'll do

• Architect, design, and develop AI/ML compiler and runtime software for RISC-V based IP, NPU, and SoC platforms
• Develop and enhance IREE-based compiler flows, including MLIR lowering, code generation, runtime integration
• Work across AI framework import paths including PyTorch, ONNX, and TFLite, and enable lowering through torch-mlir, TOSA, Linalg, and related MLIR dialects
• Optimize neural network workloads for edge deployment, including operator fusion, tiling, memory planning, quantization, layout transformation
• Enable efficient execution of AI models across CPU, vector, matrix, and NPU acceleration paths, balancing latency, throughput, memory footprint, and power efficiency

What you need

• 3–12 yrs of experience
• Machine learning
• Python
• PyTorch
• TensorFlow
• FPGA
• Linux
• Any graduate

Good to have

• LLMs
• CI/CD
• JIRA
• Prototyping
• Jenkins
• Git