Compiler Engineer-MLIR Job at DensityAI, Mountain View, CA

  • DensityAI
  • Mountain View, CA

Job Description

Job Description

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

California pay range

$240,000—$275,000 USD

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.

 

Job Tags

H1b, Work visa

Similar Jobs

Skytop Lodge

Pool Attendant Job at Skytop Lodge

 ...Pool Attendant Maintain a safe, hygienic, pleasant environment in the swimming areas. Monitor all activity and conditions for safety hazards, unsafe behavior, and medical emergencies. Take any corrective action required, following established guidelines. Perform the... 

Quality Life Services

Agency Certified Nursing Assistant (CNA) Job at Quality Life Services

 ...Staffing Solutions offers a rewarding career where you can make a difference in the lives of older adults. We are seeking an Agency Certified Nursing Assistant (CNA) who is looking for the opportunity to earn great wages while also making a profound impact in their... 

Giant Eagle

Warehouse Order Picker Job at Giant Eagle

Join Giant Eagle as a Warehouse Order Picker in our Warehouse & Manufacturing team. In this vital role, youll accurately pick, pack, and stage grocery orders using handheld scanners and pallet jacks to support our busy retail locations. Youll follow safety and food-handling... 

Emory University

Assistant Director for Spanish Programming and Events, Aquinas Center of Theology Job at Emory University

 ...collaborative Assistant Director of Spanish Programming and Events. This employee will work with the Executive Director to...  ...and initiatives among the Spanish speaking groups throughout the Archdiocese...  ...location. Remote work from home day options may be granted at department... 

University of Maine

Head Softball Coach Job at University of Maine

 ...and track academic eligibility regulations for student athletes Manages the care of team equipment, uniforms and training gear Assists...  ...relationships, and fund-raising opportunities To collaborate with the Sports Information Director and Media Relations office in the...