AI Compiler Engineer

Job Summary

To enhance AI model performance, the remote AI Compiler Engineer will architect, design, and implement optimizations for graph compilers, collaborating with hardware architects and AI researchers on cutting-edge AI and ML workloads.

Key responsibilities

• Architect and implement optimizations for AI model execution on graph compilers to improve performance and reduce latency
• Collaborate with ML researchers and hardware engineers to design and deploy AI models, addressing hardware-specific challenges
• Research and integrate advancements in compiler design and ML model optimizations into graph compilers

Required 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 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
• Proficiency in C++, Python, or other programming languages used in compiler development