Densityai

Densityai

Compiler Engineer — MLIR

Mountain View, CA

Work authorization required$200k-$360kDetected 21 days ago
C++Machine LearningTensorFlowPyTorchRecruiting

About the role

  • Work with chip-design and software teams driving DensityAI's AI accelerator program from first silicon through scale-out.

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

Requirements

  • Strong C++ fluency and experience integrating with ML frameworks (PyTorch, JAX, ONNX, TensorFlow, or equivalent)
  • Work authorization is required at start
  • Work authorization is required at start; we provide immigration support to secure or transfer status.

Compensation

  • Full compensation packages are based on candidate experience and relevant certifications.
  • California pay range
  • $200,000 - $360,000 USD

Benefits

  • Additional compensation: equity grant per company guidelines
  • medical / dental / vision

Company info

  • What we're looking for

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
  • $200,000 - $360,000 USD

Visa & Work Authorization

  • 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.
Apply directly at Densityai →Create a free account for alerts like thisView Densityai immigration profile

This listing is sourced directly from Densityai's careers page and normalized into a canonical job model.