Compiler Architect for Heterogeneous AI Cloud

Gimlet Labs; San Francisco (CA); On-site; USD 120,000 - 150,000; Full time

Job summary

A technology company specializing in AI infrastructure is seeking a Member of Technical Staff to design and implement compiler systems. This role involves building compiler pipelines for heterogeneous hardware while ensuring performance and scalability. The ideal candidate should have solid software engineering skills and experience in compiler systems, particularly with frameworks like LLVM or similar. This position is focused on enhancing AI workload execution efficiency to support advanced infrastructure.

Qualifications

• Experience building or working on compiler systems or compiler-adjacent infrastructure.
• Strong problem-solving skills and attention to detail.
• Ability to work close to hardware systems.

Responsibilities

• Design and implement compiler pipelines for heterogeneous hardware.
• Develop multi-level intermediate representations (IRs).
• Implement partitioning and lowering strategies for workloads.
• Support both ahead-of-time and JIT compilation paths.
• Integrate new model architectures into the compiler stack.
• Ensure correctness and performance across the stack.

Skills

Strong software engineering fundamentals

Experience with compiler systems

Comfort with execution and performance

C++ development experience

Python development experience

Tools

LLVM

MLIR

TVM

XLA

Job description

A technology company specializing in AI infrastructure is seeking a Member of Technical Staff to design and implement compiler systems. This role involves building compiler pipelines for heterogeneous hardware while ensuring performance and scalability. The ideal candidate should have solid software engineering skills and experience in compiler systems, particularly with frameworks like LLVM or similar. This position is focused on enhancing AI workload execution efficiency to support advanced infrastructure.