Kernel Engineer - New Grad

Cerebras Cerebras · Semiconductors · US and Canada Offices · University Departments

The Kernel Engineer role at Cerebras focuses on developing and optimizing high-performance software, specifically machine learning and linear algebra kernels, for the Cerebras Wafer-Scale Engine. This involves low-level programming, parallel algorithm application, performance analysis, and validation to maximize compute utilization and system performance for AI and HPC workloads.

What you'd actually do

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

Skills

Required

  • C++
  • Python
  • foundational computer architecture concepts
  • data structures
  • algorithms
  • software development fundamentals
  • debugging software
  • 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

Nice to have

  • kernel development
  • compilers
  • computer architecture
  • HPC
  • systems programming
  • parallel algorithms
  • multithreaded programming
  • distributed memory systems
  • programming accelerators (GPUs, FPGAs)
  • low-level programming
  • assembly language
  • CUDA
  • OpenCL
  • domain-specific language
  • machine learning concepts
  • neural networks
  • PyTorch
  • TensorFlow
  • numerical computing
  • linear algebra
  • HPC kernels
  • profiling tools
  • benchmarking tools
  • performance analysis tools
  • library or API development practices

What the JD emphasized

  • high-performance software
  • machine learning and linear algebra kernels
  • Cerebras Wafer-Scale Engine
  • low-level programming
  • parallel programming algorithms
  • performance analysis
  • kernel development
  • compilers
  • computer architecture
  • HPC
  • systems programming
  • low-level programming
  • performance optimization
  • hardware/software co-design
  • machine learning concepts
  • neural networks
  • performance analysis tools

Other signals

  • Develop high-performance software at the intersection of hardware and software for cutting-edge artificial intelligence and high-performance computing workloads.
  • Implement, optimize, and validate machine learning and linear algebra operations for the Cerebras Wafer-Scale Engine.
  • Design, development, performance tuning, and validation of foundational ML and HPC kernels.