Compiler Engineer — MLIR
DensityAI•Mountain View, CA
•$200,000 - $360,000•Remote
About The Position
This role involves access to ITAR-controlled information. Applicants must be U.S. persons (U.S. citizens, U.S. permanent residents, asylees, or refugees) per 22 CFR 120.62. The role requires owning 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. The engineer will 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
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
Upload and Match Resume
What This Job Offers
Job Type
Full-time
Career Level
Senior
Education Level
No Education Listed
