Front End Compiler

Posted Yesterday
Hiring Remotely in Cañada De Pala, CA, USA
In-Office or Remote
Junior
Artificial Intelligence • Machine Learning • Software
The Role
Build and maintain the front end of a portable AI compiler: parsing, validation, and IR lowering from frameworks (PyTorch, StableHLO, ONNX, MLIR) into internal IR. Ensure operator coverage, collaborate with optimization and codegen teams, and improve correctness, tooling, and ingestion pipelines.
Summary Generated by Built In
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
RequirementsEssential 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.

Skills Required

  • BS degree in Computer Science or equivalent practical experience
  • 2+ years working with ML optimization tools/libraries
  • Experience ingesting from ML frameworks such as PyTorch, TensorFlow, JAX, or ONNX
  • 2+ years of experience with C/C++
  • 2+ years working with Python
  • Strong written and verbal communication skills; ability to write clear 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 cross-framework compatibility challenges
  • Familiarity with Python bindings and tooling for compiler front ends
  • Experience with testing infrastructure for compiler correctness (fuzzing, differential testing, model-level validation)
  • Contributions to open-source compiler or ML framework projects
  • Familiarity with MLIR, including defining and working with custom dialects
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Toronto
33 Employees
Year Founded: 2018

What We Do

At Lemurian Labs our focus is on unleashing the capabilities of AI for the benefit of humanity. To fulfill this purpose we are developing a full stack solution consisting of software and hardware that is capable of orders of magnitude better performance and efficiency than legacy solutions, while being designed for scalability. There are massive shifts underway moving us from Software 1.0 to Software 2.0 to Software 3.0 and onwards, but to realize its true benefits we need fundamentally new hardware and systems that can keep up with the changing compute demands and simultaneously bringing down costs. We are developing software and hardware designed from first principles to deliver unprecedented realizable performance/watt and enable the next generation of AI workloads. Our diverse team of technologists have decades of experience at the frontiers of high performance computing, digital arithmetic, cryptography, artificial intelligence, robotics, and networking. There is a lot of talk about what the technology of tomorrow will look like and there are a number of companies developing it. At Lemurian, we believe tomorrow is so yesterday. We are developing the technology for the day after tomorrow. We are Lemurian Labs. Welcome to the future of artificial intelligence and computing.

Similar Jobs

CertiK Logo CertiK

Front-end Engineer

Artificial Intelligence • Blockchain
Remote
US
133 Employees
140K-220K Annually

PureSpectrum Logo PureSpectrum

Sales Director, North America

Big Data • Marketing Tech • Sales • Software • Analytics • Big Data Analytics
Remote or Hybrid
USA
283 Employees

Samsara Logo Samsara

Consultant

Artificial Intelligence • Cloud • Computer Vision • Hardware • Internet of Things • Software
Easy Apply
Remote or Hybrid
United States
4000 Employees
89K-120K Annually

Similar Companies Hiring

Hanover Park Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
42 Employees
Kepler  Thumbnail
Fintech • Software
New York, New York
6 Employees
Onshore Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
60 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account