AI Compiler Engineer

A company is looking for an AI Compiler Engineer to develop and optimize graph compilers for AI and ML workloads.

Key Responsibilities

Architect, design, and implement optimizations for AI model execution on graph compilers

Collaborate with ML researchers and hardware engineers to deploy AI models and address hardware-specific challenges

Research and integrate advancements in compiler design and ML model optimizations into graph compilers

Qualifications

Bachelor's or Master's degree in Computer Science, Electrical Engineering, or related field (Ph.D. preferred)

3+ years in compiler development, focusing on AI or ML graph compilers

Proficiency in AI graph compiler frameworks (e.g., MLIR, Torch-FX)

Solid background in hardware architectures and optimization techniques

Proficiency in C++, Python, or other programming languages used in compiler development