MLIR Compiler Engineer for AI Accelerator

DensityAI; Mountain View (CA); On-site; USD 200,000 - 360,000; Full time

Benefits offered by this job

Equity grant

Medical benefits

401(k)

Standard PTO

Job summary

DensityAI in Mountain View, California, is seeking an experienced professional to own MLIR dialect design and lowering passes for their AI accelerator. You'll work closely with chip-design and software teams.

The ideal candidate has exceptional skills in MLIR design, deep knowledge of tensor compilation, and 5+ years of compiler engineering experience. The role includes a competitive compensation range of $200,000 - $360,000 annually and additional benefits.

Qualifications

• 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.

Responsibilities

• Own MLIR dialect design and lowering passes for DensityAI's AI accelerator.
• Use and develop AI-assisted tool flows to accelerate compiler development.

Skills

MLIR dialect design

Compiler engineering experience

C++ fluency

Integration with ML frameworks

Tensor compilation understanding

Tools

PyTorch

JAX

ONNX

TensorFlow

Job description

DensityAI in Mountain View, California, is seeking an experienced professional to own MLIR dialect design and lowering passes for their AI accelerator. You'll work closely with chip-design and software teams.

The ideal candidate has exceptional skills in MLIR design, deep knowledge of tensor compilation, and 5+ years of compiler engineering experience. The role includes a competitive compensation range of $200,000 - $360,000 annually and additional benefits.