Job summary
Seeking an experienced AI Compiler Engineer to develop and optimize graph compilers for advanced AI and machine learning workloads. The role involves collaboration with hardware architects and AI researchers to improve performance and enable efficient deployment on specialized inference hardware.
Location and work setup
- Location
- Norway
- Work setup
- Remote
Salary
NOK 1206458.00–1754848.00 year
Responsibilities
Design and implement compiler optimizations for AI model execution focused on enhancing performance and reducing latency. Collaborate with ML researchers, hardware engineers, and software developers to address hardware-specific challenges. Work on optimizations including layer and operator fusion, graph transformations, and generation of intermediate representations from high-level AI models. Incorporate latest advances in compiler design and hardware acceleration into graph compiler technologies. Provide technical leadership and mentorship to engineering teams.
Qualifications
Bachelor's or Master's degree in Computer Science, Electrical Engineering, or related field, with a preference for PhD. Minimum of 3 years experience in compiler development emphasizing AI or ML graph compilers. Proficiency with AI compiler frameworks such as MLIR or Torch-FX. Strong understanding of hardware architectures and optimization techniques like fusion, quantization, and tiling. Familiarity with neural network operators, code parsing, semantic analysis, and IR generation. Programming skills in C++ and Python. Experience leading teams and contributing to open-source AI software is advantageous.