Principal AI Compiler Engineer

Job Responsibilities

• Own the design, implementation, and evolution of an AI compiler toolchain that leverages AI agents to seamlessly map neural networks onto NXP’s SoC platforms.
• Pioneer new graph transformations, lowering, scheduling, and codegen strategies for CPUs and custom accelerators, driven by insights from AI-powered analytics.
• Build deep integrations with leading AI frameworks (PyTorch, TensorFlow, ONNX, and more), using AI agents to rapidly onboard new model architectures and ops.
• Push the envelope on quantization, operator fusion, memory planning, and layout transformations—combining human expertise and AI-guided design for state-of-the-art results.
• Partner with hardware and software architects, kernel hackers, and AI agents to co-design next-gen compiler and accelerator features, aligning silicon and code for maximum impact.
• Diagnose and crush performance bottlenecks with AI-enabled profiling and diagnostics, relentlessly tuning for latency, throughput, and power efficiency.
• Level up validation, benchmarking, and regression pipelines by harnessing AI agents—ensuring compiler correctness and world-class performance, release after release.
• Uplevel the developer experience by streamlining usability, diagnostics, and documentation—AI agents are your copilots for user support, troubleshooting, and rapid iteration.

Qualifications

• MS/PhD (or equivalent experience) in Computer Science, EE, or related field
• Deep experience building AI compilers, accelerator backends, or graph optimization frameworks
• Strong expertise in graph optimization and performance optimization for NPUs or custom accelerators
• Experience with MLIR, LLVM, TVM-like systems, or proprietary compiler IRs
• Excellent C/C++ and Python skills
• Solid understanding of AI inference workloads (CNNs, transformers, perception or generative models)
• Strong communication skills are required, e.g. agile development experience in Scrum team (Product Owner or Scrum Master)