Densityai

Densityai

Kernel Engineer (Compute / Accelerator)

Mountain View, CA

Work authorization required$260k-$320kDetected 21 days ago
PythonC++Machine LearningRecruitingFPGAVerilog

About the role

  • You will write, evaluate, and profile specialized compute kernels that run on a custom AI accelerator.

Responsibilities

  • Write and optimize compute kernels for a custom AI accelerator - tensor operations, data movement patterns, memory hierarchy exploitation
  • Develop and maintain profiling infrastructure to measure kernel performance against architectural targets
  • Define and document shuffle patterns for ML kernel primitives across CPU-like control, tensor cores, and CUTLASS-style operations
  • Drive kernel DSL design decisions - thread spawn mechanisms, register passing conventions, and memory management strategies
  • Collaborate with the compiler team on the MLIR dialect - your kernels are the primary validation target
  • Create onboarding documentation and kernel writing guides for the broader team

Requirements

  • (Optional) FPGA or Verilog/SystemVerilog (ability to read RTL and reason about the hardware you're targeting)

Compensation

  • Final offers depend on level, location, and skills relevant to the role.
  • 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.
  • Visa Sponsorship
  • We comply with the California CROWN Act and provide reasonable accommodations on request.
  • Full compensation packages are based on candidate experience and relevant certifications.

Benefits

  • equity grant per company guidelines; medical / dental / vision; 401(k); standard PTO.

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
  • $260,000 - $320,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.