Senior NPU Kernel & Operator Performance Architect

Black Sesame Technologies Inc; San Jose (CA); Presencial; USD 120 000 - 160 000; Tempo integral

Resumo da oferta

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.

Qualificações

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

Responsabilidades

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

Conhecimentos

Performance optimization

Accelerator programming

GPU development

NPU programming

HPC systems

Formação académica

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

Ferramentas

CUDA

Triton

OpenCL

Descrição da oferta de emprego

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.