AI/ML Compiler & Runtime Software Engineer

About the role

AI SDK Team builds the foundational software stack that enables Physical AI workloads on next‑generation RISC‑V IP and SoC platforms. The Senior Staff Engineer designs and develops AI/ML compiler and runtime software, creates custom MLIR dialects, integrates IREE flows, and optimizes models for edge accelerators. Works on‑site in Pune, collaborating with hardware, firmware and product teams to bring AI workloads from silicon to software.

What you’ll do

• Architect and develop AI/ML compiler and runtime for RISC‑V based IP
• Create and maintain custom MLIR dialects and compiler passes
• Integrate IREE‑based flows and enable framework imports
• Optimize neural network workloads for edge deployment
• Collaborate with hardware and firmware teams for silicon bring‑up
• Analyze performance and drive compiler/runtime optimizations
• Define software architecture and technical direction for AI SDK
• Build test infrastructure, benchmarks and CI pipelines

What you’ll bring

• 3-12 years software engineering
• Strong experience with LLVM/MLIR/IREE
• Developed custom compiler passes and dialects
• Proficient in C++ and Python
• Worked with AI model frameworks (PyTorch, ONNX, TFLite)
• Experience optimizing for NPUs and edge accelerators
• Linux cross‑compilation and debugging
• Technical leadership and mentorship

Skills