Compiler Code Gen Engineer

Lemurian Labs is reimagining the foundations of computing to make AI accessible to everyone. They are seeking a Compiler Code Generation Engineer to design and build core code generation capabilities inside their high-performance AI compiler, focusing on translating high-level ML computations into optimized machine code across various hardware targets.

Responsibilities:

• Design, develop, maintain, and improve our heterogeneous AI compiler
• Design and implement new code generation capabilities based on our novel compiler architecture
• Propose improvements and extensions to our compiler architecture in response to advances in ML model design and hardware
• Apply the latest techniques in parallelization and partitioning to automate kernel generation and exploit highly optimized execution paths
• Use performance data to identify optimization opportunities and drive measurable improvements
• Collaborate with our product team to understand the evolving needs of ML engineers and translate those needs into architectural improvements