Kernel Engineer - New Grad

Job summary

Responsibilities

As a Kernel Engineer at Cerebras, you will develop high-performance software at the intersection of hardware and software for cutting-edge artificial intelligence and high-performance computing workloads.

• Help design and implement machine learning and linear algebra kernels for the Cerebras Wafer-Scale Engine.
• Develop and debug high-performance kernel routines using low-level programming techniques and the Cerebras Software Language, a custom C-like language.
• Apply parallel programming algorithms to map computational workloads efficiently onto the Cerebras architecture.
• Use mathematical analysis, performance data, and profiling tools to evaluate kernel behavior and inform design decisions.
• Identify and investigate correctness, performance, and hardware utilization issues.

What they're looking for

• Bachelor’s, Master’s, or PhD in Computer Science, Computer Engineering, Electrical Engineering, Mathematics, or a related field.
• Strong programming fundamentals in C++ and familiarity with Python.
• Understanding of foundational computer architecture concepts such as processors, memory hierarchies, instruction execution, or data movement.
• Knowledge of data structures, algorithms, and software development fundamentals.
• Experience debugging software through coursework, internships, research, co-op placements, or technical projects.

Tech stack

• C++
• Python
• CUDA
• PyTorch
• TensorFlow