Staff AI Graph Compiler Engineer: Lead Optimizations

Mulya Consulting; United States; Hybrid; USD 42,000 - 83,000; Full time

Job summary

EnCharge AI is seeking an AI Compiler Engineer to lead the development and optimization of graph compilers for cutting-edge AI/ML workloads. You will work with hardware architects, researchers, and software teams to enhance performance and enable efficient model deployment on EnCharge’s Inference Accelerators.

The role requires strong expertise in AI graph compilers, extensive experience in compiler design, and a proven track record of leading engineering teams.

Qualifications

• Bachelor’s or Master’s degree in Computer Science, Electrical Engineering, or related field (Ph.D. preferred).
• 7–13 years in compiler development with a strong focus on AI/ML graph compilers.
• Proficiency in AI graph compiler frameworks (e.g., MLIR, Torch-FX).
• Solid background in hardware architectures and optimization techniques such as fusion, quantization, and tiling.
• Familiarity with neural networks operators and code generation.
• Strong understanding of intermediate representations, code parsing, and semantic analysis in compiler design.
• Proficiency in C++, Python, or other programming languages commonly used in compiler development.
• Open-source contributions to AI software frameworks is a plus.
• Demonstrated experience leading and mentoring engineering teams with successful project delivery.

Responsibilities

• Architect, design, and implement optimizations for AI model execution on graph compilers to improve performance, reduce latency, and maximize hardware utilization.
• Collaborate with ML researchers, hardware engineers, and software developers to design and deploy AI models and address hardware-specific challenges.
• Improve performance of neural networks via fusion and graph-level transformations.
• Develop compiler passes converting high-level AI models into IR for hardware accelerators.
• Implement parsing, semantic analysis, and IR generation for DL frameworks.
• Research and integrate the latest advancements in compiler design and hardware acceleration.
• Provide leadership and mentorship to a team of engineers focused on graph compiler optimizations.

Skills

Graph compiler optimization

MLIR / Torch-FX

C++ / Python

Parallel computing

Team leadership

Education

Bachelor's/Master's in CS/EE

Ph.D. preferred

Tools

MLIR

Torch-FX

Job description

EnCharge AI is seeking an AI Compiler Engineer to lead the development and optimization of graph compilers for cutting-edge AI/ML workloads. You will work with hardware architects, researchers, and software teams to enhance performance and enable efficient model deployment on EnCharge’s Inference Accelerators.

The role requires strong expertise in AI graph compilers, extensive experience in compiler design, and a proven track record of leading engineering teams.