AI Graph Compiler Engineer – Optimizing ML on Accelerators

EnCharge AI, Inc.

United States

Remote

USD 140,000 - 190,000

Full time

24 hours ago

Be an early applicant

Get a reply from this employer

Your resume and cover letter, tailored to exactly what they’re hiring for.

Get past ATS filters

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.