Kernel Engineer (Compute / Accelerator)
Most applications go out cold — see where you stand first. No sign-up to start.
Don't just apply. Show up ready.
Olive works from this exact posting — no sign-up to start.
At a glance
Job overview
DensityAI is hiring a Kernel Engineer (Compute / Accelerator). The role involves writing, evaluating, and profiling specialized compute kernels for a custom AI accelerator, serving as the critical interface between high‑level ML workloads and silicon, and collaborating closely with architecture and compiler teams to define programming models and validate hardware performance.
Key focus areas include Write and optimize compute kernels for a custom AI accelerator, Develop and maintain profiling infrastructure to measure kernel performance, and Define and document shuffle patterns for ML kernel primitives.
Successful candidates bring Work Authorization. Important skills include C/C++, CUDA, Accelerator Programming, Computer Architecture, Performance Profiling, and Performance Optimization. Preferred (not required): RISC-V, x86, ARM64 ISA, and MLIR.
Skills & qualifications
Skills
Qualifications
Benefits
Full job description
About the role You will write, evaluate, and profile specialized compute kernels that run on a custom AI accelerator. This is the critical interface between high-level ML workloads and silicon — your code directly determines how effectively the hardware performs. You'll work closely with the architecture and compiler teams to define the kernel programming model, implement core tensor operations, and drive the performance profiling workflow that validates silicon design decisions.
What you'll do
-
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
-
Enable end-to-end kernel execution on the architectural simulator
-
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
What we're looking for
-
C/C++ — production-grade systems code, not scripted glue. You'll write performance-critical kernels
-
CUDA or equivalent accelerator programming — deep experience writing GPU kernels, understanding warp/wavefront execution, memory coalescing, shared memory optimization. The mental model transfers directly
-
Computer architecture — you need to reason about pipelines, memory hierarchies, data movement costs, and how software maps to hardware
-
Performance profiling and optimization — you live in profilers. Identifying bottlenecks, measuring throughput, and iterating until kernels meet targets is the core loop
-
Tensor operations — practical understanding of GEMM, convolution, attention, reduction, and scatter/gather as they map to hardware
-
Python — for scripting, DSL integration, and profiling automation
-
(Optional) RISC-V, x86, or ARM64 ISA experience
-
(Optional) MLIR or LLVM compiler infrastructure
-
(Optional) HPC or scientific computing background (large-scale parallel compute intuition)
-
(Optional) FPGA or Verilog/SystemVerilog (ability to read RTL and reason about the hardware you're targeting)
-
(Optional) Familiarity with CUTLASS, Triton, or similar kernel libraries
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 $260,000 — $320,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.
You've read the whole posting — now see how you match it.