Kernel Engineer for AI Accelerator - Profiling

DensityAI; Mountain View (CA); On-site; USD 260,000 - 320,000; Full time

Benefits offered by this job

Medical, dental, and vision coverage

401(k) plan

Standard PTO

Equity grants

Job summary

DensityAI is seeking an expert to develop compute kernels for a specialized AI accelerator in Mountain View, California. Your role will focus on writing performance-critical kernels while collaborating with architecture and compiler teams to enhance silicon design.

Qualifications include strong proficiency in C/C++, CUDA, and performance optimization techniques. Compensation ranges from $260,000 to $320,000 USD, with additional benefits and equity grants offered.

Qualifications

• Strong experience in writing production-grade C/C++ systems code.
• Deep expertise in GPU kernels and optimization techniques.
• Understanding of computer architecture and data movement costs.

Responsibilities

• Write and optimize compute kernels for a custom AI accelerator.
• Develop profiling infrastructure to measure kernel performance.
• Collaborate with teams to define and document kernel operations.

Skills

C/C++

CUDA or equivalent accelerator programming

Computer architecture

Performance profiling and optimization

Tensor operations

Python

Job description

DensityAI is seeking an expert to develop compute kernels for a specialized AI accelerator in Mountain View, California. Your role will focus on writing performance-critical kernels while collaborating with architecture and compiler teams to enhance silicon design.

Qualifications include strong proficiency in C/C++, CUDA, and performance optimization techniques. Compensation ranges from $260,000 to $320,000 USD, with additional benefits and equity grants offered.