Front End Compiler
Lemurian Labs · Santa Clara, CA - Toronto, Canada ·
- Category
- Frontend
- Experience
- 2+ years
About Us
At Lemurian Labs, we're reimagining the foundations of computing to make AI accessible to everyone. Our mission is to remove the limits of scale, hardware, and cost that hold back innovation, so the people solving humanity's hardest problems can move faster.
We're building a new kind of software stack: a hardware-agnostic platform that makes every system — from a laptop to a supercomputer — feel like one seamless engine. Developers can write once, run anywhere, and get state-of-the-art performance across any chip, any cloud, at any scale. It's a complete rethink of how software and hardware interact — designed for the era beyond Moore's Law.
We're not looking for the comfortable or the conventional; we're looking for the bold. The engineers who crave frontier problems, who want to bend the limits of what's possible, who see infrastructure not as a constraint but as a canvas. If you want to build the foundation for the next era of AI and change what humanity can achieve in the process, join us.
About the Role
We're looking for a Front End Compiler Engineer to own the ingestion layer of our high-performance, portable AI compiler. This is where the journey begins — you'll build the systems that parse, validate, and lower representations from frameworks like PyTorch, StableHLO, ONNX, and MLIR dialects into our internal compiler IR, setting the stage for everything that follows.
If you're drawn to the elegance of well-designed language frontends, care deeply about correctness and coverage, and want your work to directly enable next-generation AI models to run anywhere — this is the role for you.
What You'll Do
- Design, develop, and maintain the front end of our heterogeneous AI compiler, including parsing, validation, and IR lowering stages
- Build and extend ingestion pipelines for ML frameworks and representations including PyTorch, StableHLO, ONNX, and MLIR-based dialects
- Define and evolve the interface between external model representations and our internal compiler IR
- Ensure correctness and completeness of operator coverage across supported frameworks and hardware targets
- Collaborate with graph optimization and code generation teams to ensure clean, well-structured IR that enables downstream transformations
- Use performance and correctness data to identify gaps in coverage and drive improvements
- Contribute to documentation and tooling that helps ML engineers understand and debug the ingestion process
Requirements
Essential Skills and Experience
- BS degree in Computer Science or equivalent practical experience
- 2+ years of experienceWorking with ML optimization tools/libraries
- Experience working with or ingesting from ML frameworks such as PyTorch, TensorFlow/JAX, or ONNX
- 2+ years of experience with C/C++ and 2+ years working with Python
- Strong written and verbal communication skills; ability to write clear and concise technical documentation
- Knowledge of quantization, sparsity, or other model-level optimization techniques
Preferred Skills and Experience
- Master's or PhD in Computer Science or equivalent
- Experience with StableHLO, XLA, or other ML-specific IRs
- Knowledge of operator semantics across ML frameworks and the challenges of cross-framework compatibility
- Familiarity with Python bindings and tooling for compiler front ends
- Experience with testing infrastructure for compiler correctness — fuzzing, differential testing, or model-level validation
- Contributions to open-source compiler or ML framework projects
- Familiarity with MLIR, including defining and working with custom dialects
Why Join Lemurian Labs
- Build the entry point for an AI compiler that targets every major hardware platform
- Work at the boundary of ML frameworks and compiler infrastructure — a rare and high-leverage intersection
- Tackle deep correctness and coverage problems that directly impact what models can run and where
- Competitive compensation including equity, medical/dental/vision, retirement savings, and wellness benefits
Lemurian Labs is an equal opportunity employer. We celebrate diversity and are committed to creating an inclusive environment for all employees, regardless of gender identity, race, ethnicity, sexual orientation, disability status, age, or background.
Compensation depends on experience and geographic location and will be narrowed during the interview process. Additional benefits include equity, company bonus opportunities, medical, dental, and vision coverage, a retirement savings plan, and supplemental wellness benefits.