Compiler Optimization Engineer

Lemurian Labs is focused on reimagining the foundations of computing to enhance AI accessibility. The Graph Optimization Compiler Engineer will be responsible for optimizing the AI compiler stack, focusing on transforming high-level model graphs for efficient code generation and improving performance outcomes.

Responsibilities:

• Design, develop, and maintain the graph optimization layer of our heterogeneous AI compiler
• Implement and extend graph-level transformation passes including operator fusion, layout propagation, dead code elimination, constant folding, and algebraic simplification
• Define and evolve our intermediate representation (IR) to support new optimization opportunities as ML model architectures advance
• Analyze performance data to identify optimization gaps and drive measurable improvements in throughput and latency
• Collaborate with front end and code generation teams to ensure clean IR interfaces and well-structured optimization pipelines
• Propose and prototype new optimization strategies in response to advances in model design and hardware capabilities
• Contribute to testing and validation infrastructure to ensure optimization correctness across model types and hardware targets