New Grad Kernel Engineer for AI HPC Systems

Cerebras; United States; On-site; USD 120,000 - 180,000; Full time

Job summary

Cerebras Systems is seeking a Kernel Engineer to develop high-performance software at the intersection of hardware and software for AI and HPC workloads. You will map ML and linear algebra operations to the Cerebras Wafer-Scale Engine, optimizing compute utilization and system performance.

You will collaborate with experienced kernel, compiler, performance, and hardware engineers, learn architecture details, and contribute to kernel libraries and validation methodologies.

Qualifications

• Bachelor’s, Master’s, or PhD in CS, CE, EE, math or related field.
• Strong C++ fundamentals and Python familiarity.
• Understanding of computer architecture concepts.
• Knowledge of data structures, algorithms, and software fundamentals.
• Experience debugging software through coursework, internships, research, co-op placements, or technical projects.
• Strong analytical and problem-solving skills.
• Interest in low-level software, parallel computing, performance optimization, or hardware/software co-design.
• Ability to learn unfamiliar systems and collaborate effectively within a technical team.

Responsibilities

• Design and implement ML and linear algebra kernels for Cerebras Wafer-Scale Engine.
• Develop and debug high-performance kernel routines using the Cerebras Software Language.
• Apply parallel programming to map workloads onto Cerebras architecture.
• Use analysis, data, and profiling tools to evaluate kernel behavior and guide design.
• Investigate correctness, performance, and hardware utilization issues.
• Develop unit tests and system validations for kernel libraries.
• Collaborate with kernel, compiler, performance, and hardware engineers to improve software.
• Study ML workloads and evolve the kernel library.
• Participate in code reviews and software development processes.
• Build understanding of Cerebras architecture and memory system.

Skills

C++

Python

Arch concepts

Data structures

Debugging

Parallel computing

Co-design

Team collaboration

Education

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

Tools

Cerebras Software Language

CUDA

OpenCL

Assembly

Job description

Cerebras Systems is seeking a Kernel Engineer to develop high-performance software at the intersection of hardware and software for AI and HPC workloads. You will map ML and linear algebra operations to the Cerebras Wafer-Scale Engine, optimizing compute utilization and system performance.

You will collaborate with experienced kernel, compiler, performance, and hardware engineers, learn architecture details, and contribute to kernel libraries and validation methodologies.