Compiler Engineer — MLIR

DensityAI • Mountain View, CA

1d

About The Position

Own MLIR dialect design and lowering passes for our AI accelerator — defining the high-level tensor IR, async / streaming semantics, and sharded-tensor types that bridge ML frameworks to silicon. Work with chip-design and software teams driving DensityAI's AI accelerator program from first silicon through scale-out.

Requirements

• Exceptional abilities in MLIR dialect design, lowering pass authoring, and rewrite patterns
• 5+ years compiler engineering experience, with hands-on MLIR contributions or equivalent IR-design experience
• Deep understanding of tensor compilation, distributed / sharded execution, and async / streaming dataflow models
• Strong C++ fluency and experience integrating with ML frameworks (PyTorch, JAX, ONNX, TensorFlow, or equivalent)

Nice To Haves

• LLVM backend experience
• GPU compiler experience (Triton, IREE, XLA, or equivalent)
• open-source MLIR / LLVM contributions

Responsibilities

• Own MLIR dialect design and lowering passes for our AI accelerator — defining the high-level tensor IR, async / streaming semantics, and sharded-tensor types that bridge ML frameworks to silicon
• Use and develop AI-assisted tool flows to accelerate compiler development

Benefits

• equity grant per company guidelines
• medical / dental / vision
• 401(k)
• standard PTO

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Education Level: No Education Listed