Senior Kernel Optimization Engineer

Cerebras; Sunnyvale (CA); On-site; USD 180,000 - 240,000; Full time

Job summary

Cerebras Systems in Sunnyvale, CA is seeking a Kernel Engineer to push the limits of high-performance ML and HPC kernels on our massively parallel processor. You will design, implement, and optimize low-level kernels and a CSL/C++-Python stack that drives real-time AI workloads on the Cerebras WSE hardware.

You will partner with hardware and software teams to tune instruction sets, validate performance, and scale kernels for state-of-the-art neural networks.

Qualifications

• Bachelor’s, Master’s, or PhD in CS/CE/Math or related field.
• Strong knowledge of C++ and Python programming.
• Experience debugging complex software stacks.
• Willingness to learn new hardware architectures.

Responsibilities

• Develop kernels and mapping to Cerebras WSE System using parallel algorithms.
• Build and debug kernel libraries in low-level assembly and CSL.
• Measure performance and inform design decisions with models.
• Create unit and system tests for kernel libraries.
• Collaborate with chip and system architects to optimize IO and microarchitecture.
• Stay updated on ML trends to evolve kernel library architecture.

Skills

C++

Python

debugging

hardware architecture concepts

library/API development

Education

Bachelor’s, Master’s, PhD in CS/CE/Math

Job description

Cerebras Systems in Sunnyvale, CA is seeking a Kernel Engineer to push the limits of high-performance ML and HPC kernels on our massively parallel processor. You will design, implement, and optimize low-level kernels and a CSL/C++-Python stack that drives real-time AI workloads on the Cerebras WSE hardware.

You will partner with hardware and software teams to tune instruction sets, validate performance, and scale kernels for state-of-the-art neural networks.