AI Compiler Engineer

EnCharge AI

$190K — $255K *
US-Anywhere
+ 2 other locationsRemote
Information Technology
Less than 5 years of experience
Job Overview by Ladders

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 (e.g., GPUs, TPUs, ASICs)
  • Familiarity with neural network operators and code generation
  • Strong understanding of intermediate representations, code parsing, and semantic analysis in compiler design
  • Proficiency in C++, Python, or other languages used in compiler development

Responsibilities

  • Architect, design, and implement optimizations for AI model execution on graph compilers
  • Collaborate with ML researchers, hardware engineers, and software developers on AI model deployment
  • Optimize performance for neural network models through techniques like layer fusion
  • Develop compiler optimizations that convert high-level AI models into intermediate representations
  • Implement parsing, semantic analysis, and IR generation for deep learning frameworks
  • Research advancements in compiler design and hardware acceleration
  • Provide leadership and technical guidance to a team focusing on graph compiler optimizations

Benefits

  • Collaborative environment with cross-functional teams
  • Leadership opportunities in a rapidly advancing field
  • Access to cutting-edge AI and ML technologies
  • Chance to work on impactful projects within AI model deployment
  • Potential for involvement in open-source contributions
Full Job Description
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.

The salary range for this position is $190,000 to $255,000 USD per year. (Per Year: $195,000 to $265,000 CAD | €110,000 to €160,000 EUR | 1,206,458 to 1,754,848 NOK)
Actual compensation offered will be determined based on job-related knowledge, skills, and experience.

Similar Jobs

More Jobs at EnCharge AI

More Information Technology Jobs

Find similar AI Compiler Engineer jobs: