← All roles

Staff Kernel Optimzation Engineer

Cerebras· Software Engineering
Apply on company site
RemoteFull-timePosted 3mo ago

About the role

Kernel engineer focused on developing, optimizing, and scaling high-performance ML and HPC kernels for Cerebras’ massively parallel processor architecture, with emphasis on low-level assembly, CSL, and C++/Python tooling to maximize compute utilization and training efficiency.

Responsibilities

  • Develop design specifications for new machine learning and linear algebra kernels and mapping to the Cerebras WSE System using various parallel programming algorithms.
  • Develop and debug kernel library of highly optimized low level assembly instruction and C-like domain specific language routines to implement algorithms targeting the Cerebras hardware system.
  • Develop and debug high-performance kernel routines in low-level assembly and a custom C-like (CSL) language, implementing algorithms optimized for the Cerebras hardware system.
  • Using mathematical models and analysis to measure the software performance and inform design decisions.
  • Develop and integrate unit and system testing methodologies to verify correct functionality and performance of kernel libraries.
  • Study emerging trends in Machine Learning applications and help evolve Kernel library architecture to address computational challenges of the start-of-the-art Neural Networks.
  • Interact with chip and system architects to optimize instruction sets, microarchitecture, and IO of next generation systems.

Qualifications

  • Bachelor’s, Master’s, PhD or foreign equivalents in Computer Science, Computer Engineering, Mathematics, or related fields.
  • Understanding of hardware architecture concepts — must be comfortable learning the details of a new hardware architecture.
  • Skilled in C++ and Python programming languages.
  • Good knowledge of library and/or API development best practices.
  • Strong debugging skills and knowledge of debugging complex software stack.

Skills

Required:C/C++Python
Nice to have:Kernel developmentParallel algorithmsAccelerator programmingMachine Learning frameworks

Benefits

  • Not specified in the description
Last updated Aug 30, 2026. Interested? Apply on the company site.

About Cerebras

Cerebras is the go-to platform for fast and effortless AI training. Learn more at cerebras.ai.

Cerebras is the go-to platform for fast and effortless AI training.

AICerebras Wafer-Scale Enginechipshardwareinferenceliquid-ai-recruiting-targetmultimodalsetter-30
Company profileWebsite

More open roles at Cerebras