AI Compiler Engineer U.S., Canada, Germany, Norway, India

EnCharge AI, Inc.; United States; Remote; USD 140,000 - 190,000; Full time

Job summary

EnCharge AI in the United States seeks an experienced AI Compiler Engineer to lead the development and optimization of graph compilers for cutting-edge AI/ML workloads. You will collaborate with hardware architects, researchers, and software engineers to enhance performance and enable efficient model deployment on EnCharge’s Inference Accelerators.

The role emphasizes hands-on compiler work, leadership, and staying at the forefront of MLIR and related frameworks to push the capabilities of

Qualifications

• 3+ years in compiler development with a focus on AI/ML graphs.
• Experience with AI graph compiler frameworks such as MLIR and Torch-FX.
• Strong background in hardware architectures and optimization techniques including fusion, quantization, and tiling.
• Proven experience leading and mentoring engineering teams.

Responsibilities

• Architect, design, and implement optimizations for AI model execution on graph compilers to improve performance and reduce latency.
• Collaborate with ML researchers, hardware engineers, and software developers to deploy AI models on EnCharge hardware.
• Perform performance optimizations for neural networks, such as layer fusion and operator fusion.
• Develop compiler passes to transform high-level models into intermediate representations (IR).
• Research and integrate advances in compiler design and ML model optimizations into graph compilers.
• Provide leadership and mentorship to a team focused on graph compiler optimizations.

Skills

AI graph compilers

C++/Python

Hardware optimization

Leadership/mentoring

Education

Bachelor’s or Master’s degree in CS/EE

Ph.D. preferred

Tools

MLIR

Torch-FX

LLVM

Job description

EnCharge AI is a leader in advanced AI hardware and software systems for edge-to-cloud computing. EnCharge’s robust and scalable next-generation in-memory computing technology provides orders-of-magnitude higher compute efficiency and density compared to today’s best-in-class solutions. The high-performance architecture is coupled with seamless software integration and will enable the immense potential of AI to be accessible in power, energy, and space constrained applications. EnCharge AI launched in 2022 and is led by veteran technologists with backgrounds in semiconductor design and AI systems.

About the Role

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

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.

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, 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 and libraries is a plus
• Demonstrated experience leading and mentoring engineering teams with successful project delivery.

EnchargeAI is an equal employment opportunity employer in the United States.