Compiler Engineer — MLIR

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

About the Role

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.

What You’ll Do

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

What We’re Looking For

• 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).
• (Optional) LLVM backend experience, GPU compiler experience (Triton, IREE, XLA, or equivalent), or open‑source MLIR / LLVM contributions.

Compensation

Final offers depend on level, location, and skills relevant to the role. Additional compensation includes equity grant per company guidelines; medical/dental/vision; 401(k); standard PTO.

Visa Sponsorship

DensityAI sponsors qualified candidates for H-1B, O-1, TN, E-3, and other employment‑based visas, and we welcome applicants on F-1 OPT and STEM‑OPT. Work authorization is required at start; we provide immigration support to secure or transfer status.

Equal Opportunity

DensityAI is an Equal Opportunity Employer. We do not discriminate on the basis of race, color, religious creed, national origin, ancestry, physical or mental disability, medical condition, genetic information, marital status, sex, gender, gender identity, gender expression, age (40+), sexual orientation, military or veteran status, pregnancy, or any other status protected by law. We comply with the California CROWN Act and provide reasonable accommodations on request.

California Pay Range

$200,000 - $360,000 USD