AI Graph Compiler Architect

Rozgoo; United States; Remote; USD 140,000 - 200,000; Full time

Job summary

EnCharge AI is seeking a highly skilled AI Compiler Engineer to lead graph compiler optimizations for cutting-edge AI/ML workloads and accelerators. You will collaborate with hardware architects and ML researchers to boost performance and deployment efficiency on EnCharge Inference Accelerators.

The role emphasizes optimizing computation graphs, implementing IR generation, and delivering scalable compiler solutions with leadership across an engineering team.

Qualifications

• Bachelor’s or Master’s degree in Computer Science, Electrical Engineering, or related field (Ph.D. preferred).
• 3+ years in compiler development, with a strong focus on AI or ML graph compilers.
• Proficiency in AI graph compiler frameworks (e.g., MLIR, Torch-FX)
• Solid background in hardware architectures (e.g., GPUs, TPUs, ASICs) and optimization techniques such as fusion, quantization, and tiling.
• Familiarity with neural networks operators and code generation.
• Strong understanding of intermediate representations.

Responsibilities

• Architect, design, and implement optimizations for AI model execution on graph compilers to improve performance, reduce latency, and maximize hardware utilization.
• Work closely with ML researchers, hardware engineers, and software developers to design and deploy AI models, understanding and addressing hardware-specific challenges.
• Work on performance optimizations for neural network models, such as layer fusion, operator fusion, and graph-level transformations.
• Develop compiler optimizations and passes that convert high-level AI models (e.g., from TensorFlow, PyTorch) into intermediate representations (IR).
• Implement parsing, semantic analysis, and IR generation for deep learning frameworks.
• Research and integrate the latest advancements in compiler design, ML model optimizations, and hardware acceleration into graph compilers.
• Provide leadership, mentorship, and technical guidance to a team of engineers focused on graph compiler optimizations.

Skills

Compiler development

AI graph compilers

Hardware optimization

Neural network operators

IR/code generation

Intermediate representations

Education

Bachelor/Master in CS or EE (PhD preferred)

Tools

MLIR

Torch-FX

Job description

EnCharge AI is seeking a highly skilled AI Compiler Engineer to lead graph compiler optimizations for cutting-edge AI/ML workloads and accelerators. You will collaborate with hardware architects and ML researchers to boost performance and deployment efficiency on EnCharge Inference Accelerators.

The role emphasizes optimizing computation graphs, implementing IR generation, and delivering scalable compiler solutions with leadership across an engineering team.