Front End Compiler

Lemurian Labs Inc.; Santa Clara (CA); On-site; USD 140,000 - 190,000; Full time

Benefits offered by this job

Equity

Medical/Dental/Vision

Retirement savings plan

Wellness benefits

Job summary

Lemurian Labs Inc. seeks a Front End Compiler Engineer to own the ingestion layer of a high-performance, portable AI compiler. You will parse, validate, and lower representations from PyTorch, StableHLO, ONNX, and MLIR dialects into our internal IR, enabling next‑gen AI models to run anywhere.

You will design the frontend, extend ingestion pipelines, ensure operator coverage, and collaborate with optimization teams to produce robust, verifiable IR and tooling for ML engineers.

Qualifications

• BS degree or equivalent practical experience in CS or related field.
• 2+ years working with ML optimization tools and libraries.
• Experience ingesting from ML frameworks such as PyTorch, TensorFlow/JAX, or ONNX.
• 2+ years in C/C++ and 2+ years in Python.
• Strong documentation and communication skills.
• Knowledge of quantization, sparsity, or model-level optimization.

Responsibilities

• Design, develop, and maintain the frontend 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

ML optimization tools

ML framework ingestion

C/C++

Python

Documentation

Model optimization techniques

Education

BS in Computer Science or equivalent

MS/PhD preferred

Tools

MLIR

StableHLO

XLA

ONNX dialects

Job description

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.