- Densityai
- Mountain View, CA
- 51 days ago
- USD 200,000 – USD 360,000
Compiler Engineer — MLIR.
Before you go
Before you leave us, sign up for our email alerts
We don't do job spam, just the best digital jobs delivered straight to your inbox.
Compiler Engineer — MLIR: our view in 3 lines...
- The Role:This role is for a compiler engineer focused on MLIR-based work for an AI accelerator program.
- The Person:The person will own MLIR dialect design and lowering passes for an AI accelerator and help develop AI-assisted tool flows to speed up compiler development.
- Requirements:The ideal candidate has exceptional abilities in MLIR dialect design, lowering pass authoring, rewrite patterns, 5+ years compiler engineering experience, strong C++ fluency, and experience with PyTorch, JAX, ONNX, or TensorFlow.
About the role
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: 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.
Full compensation packages are based on candidate experience and relevant certifications.
Compensation
Final offers depend on level, location, and skills relevant to the role. Additional compensation: 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.
Export Controls
Aspects of this role may involve access to information subject to U.S. export controls (EAR/ITAR). We may discuss licensing or scope adjustments during the interview.
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.
