Note: The job is a remote job and is open to candidates in USA. 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 for their high-performance AI compiler, 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
Skills
- BS degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent practical experience
- 4+ years of experience working with compilers
- Deep knowledge of compiler algorithms and data structures
- Experience with and genuine interest in low-level code generation, object file manipulation, and target-specific optimizations
- 4+ years of experience with C/C++
- Strong written and verbal communication skills; ability to write clear and concise technical documentation
- Master's or PhD in Computer Science, Computer Engineering, Electrical Engineering, or equivalent
- Knowledge of traditional compiler techniques: instruction selection, register allocation, dominance analysis, def-use chains
- Familiarity with calling conventions, APIs, linking, and relocations
- Working knowledge of LLVM
- Experience with loop optimizations, vectorization, unrolling, fusion, and parallelization
- Experience with machine learning workloads and their hardware demands
Benefits
- Equity
- Medical/dental/vision
- Retirement savings
- Wellness benefits
- Equity
- Company bonus opportunities
- Medical, dental, and vision coverage
- A retirement savings plan
- Supplemental wellness benefits
Company Overview
Company H1B Sponsorship