Front End Compiler

Summary

Lemurian Labs is reimagining the foundations of computing to make AI accessible to everyone. They are seeking a Front End Compiler Engineer to own the ingestion layer of their high-performance, portable AI compiler, focusing on building systems that parse and validate various frameworks into their internal compiler IR.

Responsibilities

• 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

Skills

• BS degree in Computer Science or equivalent practical experience
• 2+ years of experience working 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
• 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

Qualifications

Must Haves

• BS degree in Computer Science or equivalent practical experience
• 2+ years of experience working 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

Nice to Haves

• 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

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