Staff AI Compiler Engineer

About the Role

EnCharge AI is seeking a highly skilled Staff/Principal level AI Compiler Engineer to spearhead efforts in developing and optimizing graph compilers tailored for cutting-edge AI and ML workloads. You will collaborate with hardware architects, researchers, and developers to enhance performance, optimize computation graphs, and enable efficient model deployment on EnCharge's Inference Accelerators.

Key Responsibilities:

• Architect, design, and implement optimizations for AI model execution on graph compilers to improve performance, reduce latency, and maximize hardware utilization (including fusions like layer/operator fusion).
• Work closely with ML researchers and software developers to deploy AI models while addressing hardware-specific challenges.
• Develop compiler optimizations that convert high-level AI models (TensorFlow, PyTorch) into intermediate representations (IR), covering parsing, semantic analysis, and IR generation.
• Research and integrate advancements in compiler design, model optimization, and hardware acceleration (GPU, TPU, NPU).
• Provide technical leadership and mentorship to engineering teams focused on graph compiler optimizations.