AI Compiler Engineer

About the role

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.

What you'll do

• 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

What you need

• Machine learning
• Python
• C++
• Deep learning
• TensorFlow
• PyTorch
• PhD or B.Tech or B.E. or B.Sc or BCA or M.Tech or MCA