Senior NPU Kernel & Operator Performance Architect

Black Sesame Technologies Inc; San Jose (CA); On-site; USD 120,000 - 160,000; Full time

Job summary

Black Sesame Technologies Inc is seeking a Senior NPU Kernel/Operator Engineer in San Jose, California. This role focuses on designing and optimizing high-performance kernels for a custom AI accelerator, involving deep learning operations and performance optimization.

The ideal candidate will have over 5 years of experience in accelerator programming and a strong foundation in memory hierarchy and performance-critical kernel optimization. Collaboration with hardware and model teams is essential.

Qualifications

• 5+ years of experience in performance optimization and accelerator programming.
• Deep understanding of memory hierarchy, tiling, parallelism, and bandwidth analysis.
• Experience optimizing performance-critical kernels or numerical computation.

Responsibilities

• Design and optimize high-performance NPU kernels for neural network workloads.
• Own critical operators such as attention-style kernels and normalization.
• Optimize data movement across various memory architectures.

Skills

Performance optimization

Accelerator programming

GPU development

NPU programming

HPC systems

Education

BS/MS/PhD in CS, EE, Computer Engineering, or related field

Tools

CUDA

Triton

OpenCL

Job description

Black Sesame Technologies Inc is seeking a Senior NPU Kernel/Operator Engineer in San Jose, California. This role focuses on designing and optimizing high-performance kernels for a custom AI accelerator, involving deep learning operations and performance optimization.

The ideal candidate will have over 5 years of experience in accelerator programming and a strong foundation in memory hierarchy and performance-critical kernel optimization. Collaboration with hardware and model teams is essential.