Software Engineer, Compiler (Front-end)
ABOUT THE JOB
The compiler is central to FuriosaAI's mission to build high-performance, energy-efficient AI systems. The front end is where the compiler meets the outside world. Its mission spans three areas:
- Faithful Ingestion: Translate models from external frameworks — with their evolving semantics, dynamic behaviors, and framework-specific constructs — into a precise internal representation that the rest of the compiler can reason about with confidence.
- Structural Optimization: Reshape programs at the graph level — through operator fusion, constant propagation, and shape resolution — so that downstream compilation stages receive the cleanest possible input.
- Tensor-Level Kernel Language Design: Design and evolve a programming language that enables users to directly author models optimized for FuriosaAI hardware. As the user-level interface to the compiler's internal IR and DSL, this language should maximize hardware performance while remaining intuitive for a broad range of users.
We are looking for someone who thinks in systems, designs for extensibility, and brings rigor and clarity across the stack — from model ingestion to user-facing language design.
RESPONSIBILITIES
- Design and implement the front-end pipeline that transforms models from major deep learning frameworks such as PyTorch into the compiler's internal IR.
- Develop graph-level optimizations, including operator fusion, constant folding, shape inference, and layout transformations.
- Build extensible model ingestion structures that can accommodate new architectures such as LLM, VLA, and Multimodal models, and custom operators, while maintaining consistency and correctness.
- Design and evolve a tensor-level kernel language that exposes the capabilities of the internal IR and DSL through a consistent, well-abstracted user interface.
- Establish verification mechanisms to ensure correctness throughout the translation process.
- Collaborate with software teams and langu...